如何将我的matplotlib y轴的比例更改为y ^ 2？

Question

不是将y轴设为1,2,3,4 ...，有一种方法使其变为1,4,9,16 ...？ 在网站上找不到任何内容。

这是我的代码：

import matplotlib.pyplot as plt
import numpy as np

x = np.array(range(100))
y = 2 * np.pi * (x/9.81)**(1/2)

plt.figure()

plt.plot(x, y)
plt.title("Graph")
plt.grid(True)

plt.show()

Answer 1

IIUC，您只想替换刻度线标签。在这种情况下，您可以执行以下操作。

plt.plot(x, y)
plt.title("Graph")
plt.grid(True)

yticks = np.arange(1,20,1)

plt.yticks(yticks, yticks**2)

plt.show()