#
#  Import modules
#


from qiskit import QuantumCircuit
from qiskit.primitives import StatevectorSampler
from qiskit.visualization import plot_histogram
import matplotlib.pyplot as plt

#
#  Define quantum circuit
#

qc = QuantumCircuit(1)      # |0>
qc.h(0)                     # Hadamard rotation of 45 deg  1/sqrt(2) (|0>+|1>)
qc.x(0)                     # not x                        1/sqrt(2) (-|0>+|1>)
qc.measure_all()            # still 50% 0 and 50% 1


#
# Visualize circuit and resultss
#

qc.draw("mpl")

sampler = StatevectorSampler()
result = sampler.run([qc], shots=1024).result()
counts = result[0].data.meas.get_counts()

print(counts)
plot_histogram(counts)
plt.show()
