import periodicity_1 import math
periodicity_1.plot_distribution(n_qubits=5, n_marked_states=3, n_g_max=10, beta=math.pi/4)
1/9 2/9 3/9 4/9 5/9 6/9 7/9 8/9 9/9
Try it yourself: