Man and History: Python 量子運算（六）：角度與弧度

2022年12月18日星期日

Python 量子運算（六）：角度與弧度

2022/11/28

-----

Fig. 6.1. Degree and radian.

-----

代碼 6.1

# Program 6.1：Degree and radian
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt


def DrawSubplot(N, i, j):
    splt_number = i * N + j + 1
    ax = plt.subplot(N, N, splt_number)

    # basic setting
    theta = np.linspace(0, 2*np.pi, 100)
    axis = np.linspace(-1, 1, 100)
    zero = axis * 0
    radius = 0.7
    x1 = radius*np.cos(theta)
    y1 = radius*np.sin(theta)

    # basic plotting
    ax.plot(axis, zero, 'k')  # y = 0(x_axis)
    ax.plot(zero, axis, 'k')  # x = 0(y_axis)
    ax.plot(x1, y1, 'k')      # cicle

    # vector setting
    coordinates = [0, 0]      # original point
    splt_radian = [10*(2*np.pi/360), np.pi/6, 1, np.pi/3]
    splt_title = [r"$10^{\circ}$", r"$30^{\circ}$", r"$1rad$", r"$60^{\circ}$"]
    splt_index = ['(a)', '(b)', '(c)', '(d)']

    # vector plotting
    vector_p = splt_radian[splt_number-1]
    vector = [radius*np.cos(vector_p), radius*np.sin(vector_p)]  # direction
    plt.quiver(coordinates[0], coordinates[1], vector[0], vector[1],
               scale=2, color='r')

    ax.set_aspect(1)  # height : width
    ax.set_axis_off()

    if splt_number == 2:
        ax.text(1.1, -0.15, r"$Z$")
        ax.text(0.1, 1, r"$X$")
        ax.text(0.35, 0.03, r"$\theta$", fontsize=40)
    elif splt_number == 4:
        ax.text(1.1, -0.15, r"$X$")
        ax.text(0.1, 1, r"$Y$")
        ax.text(0.2, 0.1, r"$\phi$", fontsize=40)
    else:
        ax.text(1.1, -0.15, r"$X$")
        ax.text(0.1, 1, r"$Y$")

    ax.text(radius+0.1, -0.15, r"$A$")
    ax.text(radius*np.cos(vector_p)+0.1, radius*np.sin(vector_p), r"$B$")

    plt.title(splt_title[splt_number-1], fontsize=80)
    ax.text(0.1, -1, splt_index[splt_number-1])
    return


mpl.rcParams['text.usetex'] = True
mpl.rcParams['text.latex.preamble'] = r'\usepackage{{amsmath}}'
mpl.rcParams['font.size'] = 20
fig = plt.figure(figsize=(16, 16))

N = 2
for i in range(N):
    for j in range(N):
        DrawSubplot(N, i, j)

# plt.savefig('/content/drive/My Drive/pqc/0006_001.png')
plt.show()