g2dZddlmZddlZddlmZd d ZdS)z=A collection of useful functions for post processing results.) annotationsN)make_dict_observablecountsdict observabledict | np.ndarray | listreturnfloatct|tst|}d}t|}|D]}||vr|||||z|z z }|S)aCompute the mean value of an diagonal observable. Takes in a diagonal observable in dictionary, list or matrix format and then calculates the sum_i value(i) P(i) where value(i) is the value of the observable for state i. Args: counts (dict): a dict of outcomes from an experiment observable (dict or matrix or list): The observable to be averaged over. As an example, ZZ on qubits can be given as: * dict: {"00": 1, "11": 1, "01": -1, "10": -1} * matrix: [[1, 0, 0, 0], [0, -1, 0, 0, ], [0, 0, -1, 0], [0, 0, 0, 1]] * matrix diagonal (list): [1, -1, -1, 1] Returns: Double: Average of the observable r) isinstancerrsumvalues)rrtemptotkeys Z/granary/edev/quantum/lib/python3.11/site-packages/qiskit/quantum_info/analysis/average.py average_datarsy$ j$ ' '6)*55 D fmmoo  C88 *   F3K*S/1C7 7D K)rrrr r r )__doc__ __future__rnumpynpmake_observablerrrrrs\DC""""""111111r