我无法使用matplotlib制作具有相等x和y轴的等高线图。请注意,我尝试使用相等的figsize(10,10)定义图形,并在轴函数中尝试'相等'。没有工作,正如椭圆形的“圆圈”所显示的那样:
import matplotlib.pyplot as plt
from matplotlib.colors import BoundaryNorm
from matplotlib.ticker import MaxNLocator
import numpy as np
# generate 2 2d grids for the x & y bounds
y, x = np.mgrid[slice(0, 1 + 0.1, 0.1),
slice(0, 1 + 0.1, 0.1)]
z = y + x
z = z[:-1, :-1]
levels = MaxNLocator(nbins=15).tick_values(z.min(), z.max())
# pick the desired colormap, sensible levels, and define a normalization
# instance which takes data values and translates those into levels.
plt.figure(num=None, figsize=(10,10))
cmap = plt.get_cmap('nipy_spectral')
norm = BoundaryNorm(levels, ncolors=cmap.N, clip=True)
# contours are *point* based plots, so convert our bound into point centers
plt.contour(x[:-1, :-1] + 0.1 / 2.,
y[:-1, :-1] + 0.1 / 2., z, levels=levels,
cmap=cmap, zorder=1)
plt.colorbar()
plt.axis([0, 1, 0, 1],'equal')
plt.tight_layout()
circle=plt.Circle((0.4,0.5),.1,color='k',fill=False)
plt.gca().add_artist(circle)
plt.savefig('not_square')
答案 0 :(得分:2)
要获取正方形图,请尝试plt.axis('square');
def f(x, y):
return x ** 2 + y ** 2
x = np.linspace(-2, 2, 50)
y = np.linspace(-2, 2, 50)
X, Y = np.meshgrid(x, y)
Z = f(X, Y)
plt.contour(X, Y, Z, colors='black');
plt.axis('square');