迹忆客 专注技术分享

我们可以使用 set_aspect() 方法设置绘图的纵横比以使其成为正方形绘图，而 axis() 方法也可以用于在 Matplotlib 中制作具有相等轴的正方形绘图。

set_aspect() 绘制等轴方图

我们可以使用 matplotlib.axes.Axes.set_aspect() 函数设置长宽比。如果在函数中使用等于作为长宽比，则得到的图在 X 轴和 Y 轴上从数据点到绘图单位的缩放比例相同。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3,3,100)
y = np.sin(x)

fig = plt.figure()
ax = fig.add_subplot(111)

plt.plot(x, y)
plt.xlim(-3,3)
plt.ylim(-3,3)

ax.set_aspect('equal', adjustable='box')

plt.xlabel("x")
plt.ylabel("sinx")

plt.show()

plt.xlim(-3,3)
plt.ylim(-3,3)

ax.set_aspect('equal')

它将 X 轴和 Y 轴设置为相同的范围。然后 ax.set_aspect('equal') 将两个轴设置为相等。

当两个轴的范围设置为相同时，上述方法仅产生一个正方形图。要在一般情况下生成正方形图，我们必须使用以下命令手动设置纵横比：

axes.set_aspect(1./axes.get_data_ratio())

axes.get_data_ratio() 获取原始绘图数据的比率。倒数的值传递给 set_aspect() 以使轴相等，而无需手动设置轴的限制。

代码：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3,3,100)
y = np.sin(x)

fig = plt.figure()
ax = fig.add_subplot(111)

plt.plot(x, y)

ax.set_aspect(1.0/ax.get_data_ratio(), adjustable='box')

plt.xlabel("x")
plt.ylabel("sinx")
plt.show()

使用 axis() 方法生成正方形图

如果我们将 "square" 作为参数传递给 matplotlib.pyplot.axis()，则会创建一个正方形图其中两个轴所占的范围等于图中的长度。

import numpy as np
import matplotlib.pyplot as plt

x=np.linspace(-3,3,100)
y=np.sin(x)
plt.plot(x, y)
plt.axis('square')
plt.xlabel("x")
plt.ylabel("sinx")
plt.show()