z i: SrSSKJr SSKJr SSKJrJr SSK J r JrJ r J r SSK rSSKJr SSKJrJrJrJr S S KJr S S KJrJr \ (aSS KJr S S/r\ \\S4r\ \ \\\\\ \ \4\!44r"\ \"\4r#"SS\5r$g)z5 ND-Array container class for Estimator observables. ) annotationsdeepcopy)IterableMapping)Unionroverload TYPE_CHECKINGN) ArrayLike)Pauli PauliList SparsePauliOpSparseObservable) object_array) ShapedMixin shape_tuple)TranspileLayoutObservableLikeObservablesArrayLikecl^\rSrSrSrSrSSU4Sjjjr\SSj5rSr SSjr SSSjjr S S!S jjr \ S"S j5r\ S#S j5rS r\ S$S j5r\ S#Sj5rSrS%SjrS&Sjr\S'Sj5r\S(Sj5r\S)Sj5rS*S+SjjrSrS,S-SjjrSrSrU=r$).ObservablesArray7zJAn ND-array of Hermitian observables for an :class:`.Estimator` primitive.)_array_shapec>[TU]5 [U[5(a URn[ X[ 4S9UlURRUlX l U(a[R"UR5HhupVURU5nURcURUl O%URUR:wa [S5eXpRU'Mj OTURcGURRS:a-URR!S5SRUl URcSUl gg)aInitialize an observables array. Args: observables: An array-like of basis observable compatible objects. copy: Specify the ``copy`` kwarg of the :func:`.object_array` function when initializing observables. num_qubits: The number of qubits of the observables. If not specified, the number of qubits will be inferred from the observables. If specified, then the specified number of qubits must match the number of qubits in the observables. validate: If true, coerce entries into the internal format and validate them. If false, the input should already be an array-like. Raises: ValueError: If ``validate=True`` and the input observables array is not valid. )copy list_typesNzSThe number of qubits must be the same for all observables in the observables array.r)super__init__ isinstancerrrr shaper _num_qubitsnp ndenumeratecoerce_observable num_qubits ValueErrorsizereshape) self observablesr(rvalidatendiobs basis_obs __class__s h/home/james-whalen/.local/lib/python3.13/site-packages/qiskit/primitives/containers/observables_array.pyr!ObservablesArray.__init__<s&,  k#3 4 4%,,K";ylS kk'' % NN4;;7 2237 ##+'0';';D$%%)=)==$-$- C 8   %$++*:*:Q*>#{{2226q9DDD     # D  $c0nUR5Hup#n[U5S:XaSUR-nO[[ X255n/nSnUH#upUR SX- S- -U -5 U nM% UR SUR[ U5- S- -5 SRU5SSS2n[R"U5X'M U$)zWConvert a simplified sparse observable to a mapping from Pauli strings to coefficients.rIrrN) to_sparse_listlenr(sortedzipappendmaxjoinr%real) r0resultsparse_pauli_str pauli_qubitscoefffull_pauli_str sorted_listsstring_fragments prev_qubitqubitpaulis r3 _obs_to_dictObservablesArray._obs_to_dictks585G5G5I 1 E#$)!$s~~!5%c,&IJ #%  $0LE$++C53E3I,JU,RS!&J%1!''s~~L@Q/QTU/U(VW!#)9!:4R4!@&(WWU^F "#6J& r5c[U5RS3nSURS3n[R"UR 5XSS9nX-U-$)N(z, shape=)2)prefixsuffix threshold)type__name__r#r% array2string __array__)r,rQrRarrays r3__repr__ObservablesArray.__repr__sRJ''(*DJJUR5R5$)aConvert to a nested list. Similar to Numpy's ``tolist`` method, the level of nesting depends on the dimension of the observables array. In the case of dimension 0 the method returns a single observable (``dict`` in the case of a weighted sum of Paulis) instead of a list. Examples:: Return values for a one-element list vs one element: >>> from qiskit.primitives.containers.observables_array import ObservablesArray >>> oa = ObservablesArray.coerce(["Z"]) >>> print(type(oa.tolist())) >>> oa = ObservablesArray.coerce("Z") >>> print(type(oa.tolist())) )rWtolistr,s r3r\ObservablesArray.tolists&~~&&((r5c Ub U[:XaUR[SURR555n[ URR5S:Xa3[ R"URR[S9nX4S'U$[ R"UR[S9n[ R"UR5HupVURU5XE'M U$[S5e)z6Convert to a Numpy.ndarray with elements of type dict.c38# UHn[S5v M g7fN)slice).0_s r3 -ObservablesArray.__array__..s/WEVd EVsr)r#dtypergzType must be 'None' or 'object') object __getitem__tuplerr#r:r%ndarraydictr&rKr))r,rgr tmp_resultrAr/r0s r3rWObservablesArray.__array__s =EVO))%/WT[[EVEV/W*WXJ4;;$$%*$++*;*;4H'r MJ$4$4DA "z/@/@ AHC"&"3"3C"8FK!BM:;;r5cdU(aUR5nUR$UnUR$)aConvert to a :class:`numpy.ndarray` with elements of type :class:`~.SparseObservable`. Args: copy: Whether to make a new array instance with new sparse observables as elements. Returns: A :class:`numpy.ndarray` with elements of type :class:`~.SparseObservable`. )rr)r,rr0s r3sparse_observables_array)ObservablesArray.sparse_observables_arrays,"diikzz(,zzr5cgrarhr,argss r3rkObservablesArray.__getitem__sORr5cgrarhrus r3rkrwsX[r5cURUn[U[R5(dUR U5$[ USSS9$)NFrr.)rr"r%rmrKrr,rvitems r3rkrwsA{{4 $ ++$$T* *55AAr5cgrarhrus r3rbObservablesArray.slicesFIr5cgrarhrus r3rbr~sRUr5cxURUn[U[R5(dU$[ USSS9$)a7Take a slice of the observables in this array. .. note:: This method does not copy observables; modifying the returned observables will affect this instance. Returns: A single :class:`~.SparseObservable` if an integer is given for every array axis, otherwise, a new :class:`~.ObservablesArray`. Frz)rr"r%rmrr{s r3rbr~s7{{4 $ ++K55AAr5cZ[U6n[URRU5SSS9$)zReturn a new array with a different shape. This results in a new view of the same arrays. Args: shape: The shape of the returned array. Returns: A new array. Frz)rrrr+)r,r#s r3r+ObservablesArray.reshapes-U# 3 3E :QVWWr5c8URUR5$)zReturn a new array with one dimension. The returned array has a :attr:`shape` given by ``(size, )``, where the size is the :attr:`~size` of this array. Returns: A new flattened array. )r+r*r]s r3ravelObservablesArray.ravels||DII&&r5cUR$)z-The number of qubits each observable acts on.)r$r]s r3r(ObservablesArray.num_qubitssr5c[U[5(a[R"U5nGO5[U[5(a[R "U5nGO[U[ 5(a[R"U5nO[U[5(a/nUR5Hup4[U[ 5(aURX445 M.[U[5(aCUSSR5URpeURU[SS5U-U-45 M[S[U535e [R "U5n[U[5(aUR#5n[$R&"UR(5n[$R*"U5(a [-S5e[R."UR0UUR2UR4UR65nU[R8"UR05:Xa [-S5eU$[S[U535e)a,Format an observable-like object into the internal format. Args: observable: The observable-like to format. Returns: The coerced observable. Raises: TypeError: If the input cannot be formatted because its type is not valid. ValueError: If the input observable is invalid or empty. NrrzInvalid observable basis type: zrNon-Hermitian input observable: the simplified input observable has a non-zero imaginary part in its coefficients.zEmpty observable was detected.zInvalid observable type: )r"rrfrom_sparse_pauli_opr from_paulistr from_label_Mappingitemsr=to_labelphasecomplex TypeErrorrT from_listsimplifyr% real_if_closecoeffs iscomplexobjr)from_raw_partsr( bit_termsindices boundarieszero)cls observable term_listbasisrDunphased_basisrrs r3r'"ObservablesArray.coerce_observables j- 0 0)>>zJJ  E * *)44Z@J  C ( ()44Z@J  H - -I * 0 0 2 eS))$$e^4u--,1!H,=,=,?E$$ngamu6Lu6T%UV#&Ed5k]$STT!3*33I>J j"2 3 3#,,.J%%j&7&78Fv&& ; *88%%$$""%% J-22:3H3HII !ABB 3D4D3EFGGr5c@[U[5(aU$U"U5$)zCoerce ObservablesArrayLike into ObservableArray. Args: observables: an object to be observables array. Returns: A coerced observables array. )r"r)rr-s r3coerceObservablesArray.coerce6s# k#3 4 4 ;r5cjURUR:wdURUR:wag[R"UR5n[ UR R 5UR R 55HupEXE- RU5U:wdM g g)atCompute whether the observable arrays are equal within a given tolerance. Args: other: Another observables array to compare with. tol: The tolerance to provide to :attr:`~.SparseObservable.simplify` during checking. Returns: Whether the two observables arrays have the same shape and number of qubits, and if so, whether they are equal within tolerance. FT)r(r#rrr<rrr)r,othertolzero_obsobs1obs2s r3 equivalentObservablesArray.equivalentDs ??e.. .$** 2K#((9dkk//15<<3E3E3GHJD %%c*h6Ir5c[U5$)z Return a deep copy of the array.rr]s r3rObservablesArray.copyYs ~r5cUcUcUR5$[R"UR[S9n[R "UR 5HupEURX5X4'M [USS9$)aApply a transpiler layout to this :class:`~.ObservablesArray`. Args: layout: Either a :class:`~.TranspileLayout`, a list of integers or None. If both layout and ``num_qubits`` are none, a deep copy of the array is returned. num_qubits: The number of qubits to expand the array to. If not provided then if ``layout`` is a :class:`~.TranspileLayout` the number of the transpiler output circuit qubits will be used by default. If ``layout`` is a list of integers the permutation specified will be applied without any expansion. If layout is None, the array will be expanded to the given number of qubits. Returns: A new :class:`.ObservablesArray` with the provided layout applied. Raises: QiskitError: ... riF)r.) rr%rmr#rr&r apply_layoutr)r,layoutr(new_arrr/r0s r3rObservablesArray.apply_layout]si, >j099; **TZZ/?@t{{3HC++F?GL4 %88r5cURRS5H(nURUR:wdM[S5e g)z.Validate the consistency in observables array.rz^An observable was detected, whose number of qubits does not match the array's number of qubitsN)rr+r(r))r,r0s r3r.ObservablesArray.validate|s>;;&&r*C~~0 C+r5)rr$r)NTT)r-rr( int | Nonerboolr.r)r0rreturnMapping[str, float])rzlist | ObservableLike)NN)r np.ndarray)F)rrrr)rvint | tuple[int, ...]rr)rvz!IndexType | tuple[IndexType, ...]rr)rvrrr)r#zint | Iterable[int]rr)rr)rint)rrrr)r-rrr)g:0yE>)rrrfloatrrra)rz"TranspileLayout | list[int] | Noner(rrr)rU __module__ __qualname____firstlineno____doc__ __slots__r! staticmethodrKrYr\rWrrr rkrbr+rpropertyr( classmethodr'rrrrr.__static_attributes__ __classcell__)r2s@r3rr7s9T$I "& -!)-!-! -!  -!-!^0' )* < RR [[BII UUB" X '  8H8Ht    * TX989FP9 9>r5r)%r __future__rrrcollections.abcrrrtypingrr r numpyr% numpy.typingr qiskit.quantum_infor r rrrr#rrqiskit.transpiler.layoutr__all__rrb IndexTyperrrrrrhr5r3rs#9::"QQ&+8 3 4 #ud" #   E#u* u $% 'U^Y67ZL{Lr5