g LdZddlZddlmZmZddlmZddlm Z m Z hdZ dZ dS) zDRoutines for computing expectation values from sampled distributionsN)sampled_expval_floatsampled_expval_complex) QiskitError)QuasiDistributionProbDistribution>01IZc$ddlm}ddlm}m}t |t tfr|}t ||tfstdt |tr+| g}tjdg}nt ||r+|g}tjdg}nRt ||r3|j}tj|j}ntdt't)t+|t-fd|Drtd d |D]<}t/|t2rtd |d =|jtjt6jkrt;|||St=|||S) a&Computes expectation value from a sampled distribution Note that passing a raw dict requires bit-string keys. Parameters: dist (Counts or QuasiDistribution or ProbDistribution or dict): Input sampled distribution oper (str or Pauli or PauliOp or PauliSumOp or SparsePauliOp): The operator for the observable Returns: float: The expectation value Raises: QiskitError: if the input distribution or operator is an invalid type r)Countsr)Pauli SparsePauliOpzInvalid input distribution typeg?zInvalid operator typec3>K|]}t|kVdS)N)len).0op bitstring_lens R/granary/edev/quantum/lib/python3.11/site-packages/qiskit/result/sampled_expval.py z,sampled_expectation_value..@s. 8 83r77m # 8 8 8 8 8 8z'One or more operators not same length (z) as input bitstringszInput operator z is not diagonal)countsrqiskit.quantum_inforr isinstancerrbinary_probabilitiesdictrstruppernpasarrayto_labelpaulis to_labelscoeffsrnextiteranyset differenceOPERSdtypecomplextyperr) distoperrrr oper_strsr%rrs @rsampled_expectation_valuer2s"88888888$*,<=>>+((** dVTN + +=;<<<$ 3ZZ\\N SE"" D% 3]]__% SE"" D- ( (3K))++ DK((1222T$ZZ(())M 8 8 8 8i 8 8 888  Zm Z Z Z   FF r77  e $ $ FDDDDEE E F|rx((---%i>>>#Ivt<<r9sKJ[[[[[[[[))))))>>>>>>>> 1=1=1=1=1=r