z i0SrSSKJrJrJrJrJr SSKJr SSK J r SSK r SSK Jr SSKJr SSKJr SS KJr SS KJr SS KJr SS KJr S S \\\4S\\\S\S\S\\S\\\\4\44 Sjjr Sr!S!S\\\\ RD4S\\\S\S\S\S\S\\\\ RD44Sjjr#S"S\S\\\S\S\\\44Sjjr$S#Sjr%Sr&Sr'g)$z+Utility functions for working with Results.)SequenceUnionOptionalDictList)Counter)deepcopyN) QiskitError)Result)Counts)ProbDistribution)QuasiDistribution) _bin_to_hex)resultsresultindicesinplaceformat_marginalmarginalize_memoryreturnc0[U[5(Ga^U(d [U5n[UR5GH0upVUR U5n[ Xq5n0n UR5HupX[U 5'M XRl UbO[U5URS'URRSS5n U b[X5URS'[URSS5cMUcMUSLa[!URS5 MUcM[#USS9n [$R&"URR(U SS9URlGM3 U$[ X5nU(aUb [+XU5nU$) aDMarginalize counts from an experiment over some indices of interest. Args: result: result to be marginalized (a Result object or a dict(str, int) of counts). indices: The bit positions of interest to marginalize over. If ``None`` (default), do not marginalize at all. inplace: Default: False. Operates on the original Result argument if True, leading to loss of original Job Result. It has no effect if ``result`` is a dict. format_marginal: Default: False. If True, takes the output of marginalize and formats it with placeholders between cregs and for non-indices. marginalize_memory: If True, then also marginalize the memory field (if present). If False, remove the memory field from the result. If None, leave the memory field as is. Returns: Result or dict(str, int): A Result object or a dictionary with the observed counts, marginalized to only account for frequency of observations of bits of interest. Raises: QiskitError: in case of invalid indices to marginalize over. N memory_slots creg_sizesmemoryFTreverse) return_hex) isinstancer r enumerater get_counts _marginalizeitemsrdatacountslenheaderget_adjust_creg_sizesgetattrdelattrsorted results_rsmarginal_memoryr_format_marginal)rrrrriexperiment_resultr$ new_countsnew_counts_hexkvcsizesorted_indices marg_countss M/home/james-whalen/.local/lib/python3.13/site-packages/qiskit/result/utils.pymarginal_countsr9 sx@&&!!f%F$-fnn$= A&&q)F%f6JN"((*12{1~.+,: " " )";>w<!((8)0044\4H$=OPU=_%,,\:(--x>JwOb%.-22H='/%+&N5?4N4N)..55~RV5%**1/%>4 "63 w2*6HKc UVs/sH o"SS/PM nn[U5nUVVs/sHupVUPM nnn[S[U5S-5Vs/sHn[USUS25S- PM snSSnSn UH;n [U [U55Hn XU ::dM U n X;S==S- ss' M9 M= UVs/sHo"SS:wdM UPM nnU$s snfs snnfs snfs snf)z,Helper to reduce creg_sizes to match indicesrN)r+ranger%sum) rrcregnew_creg_sizes indices_sort_x creg_nums creg_limitscreg_idxindidxs r8r(r(es0::ztAwlzN:'?L **ztqzI*6;As9~PQ?Q6RS6R3y1Q'(1,6RSTUTVWKH3{#34C#&&#A&!+& 5(6F~taAd~NF '; +SGsCCC"= C' C'r int_return hex_returnavg_dataparallel_thresholdcDU(aU(a [S5e[U[R5(aU(a [S5eU(a [S5eUcUR 5$UR S:Xa[ R"X5$UR S:Xa3U(a[ R"X5$[ R"X5$UR S:Xa[ R"X5$[S5e[ R"UUUUUS9$) a?Marginalize shot memory This function is multithreaded and will launch a thread pool with threads equal to the number of CPUs by default. You can tune the number of threads with the ``RAYON_NUM_THREADS`` environment variable. For example, setting ``RAYON_NUM_THREADS=4`` would limit the thread pool to 4 threads. Args: memory: The input memory list, this is either a list of hexadecimal strings to be marginalized representing measure level 2 memory or a numpy array representing level 0 measurement memory (single or avg) or level 1 measurement memory (single or avg). indices: The bit positions of interest to marginalize over. If ``None`` (default), do not marginalize at all. int_return: If set to ``True`` the output will be a list of integers. By default the return type is a bit string. This and ``hex_return`` are mutually exclusive and can not be specified at the same time. This option only has an effect with memory level 2. hex_return: If set to ``True`` the output will be a list of hexadecimal strings. By default the return type is a bit string. This and ``int_return`` are mutually exclusive and can not be specified at the same time. This option only has an effect with memory level 2. avg_data: If a 2 dimensional numpy array is passed in for ``memory`` this can be set to ``True`` to indicate it's a avg level 0 data instead of level 1 single data. parallel_threshold: The number of elements in ``memory`` to start running in multiple threads. If ``len(memory)`` is >= this value, the function will run in multiple threads. By default this is set to 1000. Returns: marginal_memory: The list of marginalized memory Raises: ValueError: if both ``int_return`` and ``hex_return`` are set to ``True`` z=Either int_return or hex_return can be specified but not bothz3int_return option only works with memory list inputz3hex_return option only works with memory list inputr<zInvalid input memory array) return_intrrL) ValueErrorrnpndarraycopyndimr,marginal_measure_level_1_avgmarginal_measure_level_0_avgmarginal_measure_level_1marginal_measure_level_0r-)rrrIrJrKrLs r8r-r-sTjXYY&"**%% RS S RS S ?;;= ;;! ::6K K ;;! !>>vOO!::6KK ;;! 66vG G566  % %-  r:r$c4[[UR55RSS55nUbI[U5S:Xd([ U5R [ U55(d[SUS- S35e[U[5(a[R"X5nO[U[[45(a[R"X5nO[[!UR#555n[U[$[&R(45(a[R"X5nOG[U[*[&R,45(a[R"X5nO [S5eU(aUb [/XU5$U$)a(Marginalize counts from an experiment over some indices of interest. Unlike :func:`~.marginal_counts` this function respects the order of the input ``indices``. If the input ``indices`` list is specified then the order the bit indices are specified will be the output order of the bitstrings in the marginalized output. Args: counts: result to be marginalized indices: The bit positions of interest to marginalize over. If ``None`` (default), do not marginalize at all. format_marginal: Default: False. If True, takes the output of marginalize and formats it with placeholders between cregs and for non-indices. Returns: dict(str, int): A marginalized dictionary Raises: QiskitError: If any value in ``indices`` is invalid or the ``counts`` dict is invalid.  rindices must be in range [0, r<].z(Values of counts must be an int or float)r%maxkeysreplacesetissubsetr=r rr r,r9r rmarginal_distributionnextitervaluesintrRintegerfloatfloatingr.)r$rr num_clbitsres first_values r8rdrds32S'//R89JG  1W9N9NuU_O`9a9a9*q.9ILMM&&!!((9 F-/@A B B..v?4 01 kC#4 5 5,,V=C  eR[[%9 : :226CCHI I7.W55 Jr:c [[[U55RSS55nUb![ [ U55[ U5:Xa,0nUR 5HupE[U5nXSU'M U$U(a1[ U5R[ [ U555(d[SUS- S35e[USS9n[5nUR 5HCupESRUVs/sHn[U5U*S- PM sn5nXh==U- ss'ME [U5$s snf)z4Get the marginal counts for the given set of indicesr[r\r]r<r^Tr)r%rerfrarbr=r"_remove_space_underscorercr r+rjoindict) r$rrlretkeyvalr1rHnew_keys r8r!r!s T$v,'//R89J Cj 12c'lB HC*3/CH' #g,//E*4E0FGG9*q.9ILMMWd+GJLLN''gVgs3C8#BgVWs"#  WsD> c0n[[U55n[URSS55n[ USS9nUHn[ [ Xg55nSR[U5V s/sHoU;aXOSPM sn 5SSS2n [U5HupU S:XdM U SU S-XS-n M XX:'M U$s sn f)zWTake the output of marginalize and add placeholders for multiple cregs and non-indices.r[r\TrrBN) rerfr%rar+rrziprqr=r) r$r7r format_countscounts_template counts_len indices_revcount index_dictindex count_bitsbits r8r.r.sM4<(O_,,S"56J$/K#k12 WWLQR\L] ^L]5:"5Z 3 >L] ^ B$ $O4JEcz'/#5 68JJ 5%0$6 !  _s-C cFURSS5RSS5$)z1Removes all spaces and underscores from bitstringr[r\rB)ra) bitstrings r8rprp$s"   S" % - -c2 66r:)NFFT)NFFFi)NF)N)(__doc__typingrrrrr collectionsrrTr numpyrRqiskit.exceptionsr qiskit.result.resultr qiskit.result.countsr 'qiskit.result.distributions.probabilityr !qiskit.result.distributions.quasirqiskit.result.postprocessrqiskit._acceleraterr,rrrhboolstrr9r(rSr-rdr!r.rpr:r8rs288)''D?14 $(!)- B $, B d3i BB B ! B  4S>6 !" BJ8$(" D $s)RZZ' (D d3i DD D  D  D 49bjj !DR(,!, , hsm $,, #s(^ ,^6(7r: