plot_results.py

import matplotlib.pyplot as plt


def plot_results(result, numbers, shots=1024):
    sorted_result = dict(sorted(result.items()))

    bitstrings = list(sorted_result.keys())

    # Dynamic threshold inversely proportional to the number of possible outputs
    mean_occurrence = shots / 2**len(numbers)
    threshold = mean_occurrence * 3
    
    subsets = []
    for bitstring in bitstrings:
        # bitstring is in reverse order
        indices = [i for i, bit in enumerate(bitstring[::-1]) if bit == '1']
        values = [numbers[i] for i in indices]
        subsets.append(f"{values}\n(idx: {bitstring})")

    occurrences = list(sorted_result.values())
    # Only display the labels of subsets that occur more times than the threshold
    display_labels = [label if val > threshold else "" for label, val in zip(subsets, occurrences)]

    plt.figure(figsize=(15, 9))
    plt.bar(subsets, occurrences)
    plt.xlabel("Values")
    plt.xticks(ticks=range(len(subsets)), labels=display_labels, rotation=90)
    plt.ylabel("Occurrences")
    plt.show()