我的代码输出产生2d轮廓图时遇到了一些麻烦。我在下面显示结果。
可以看出,y轴严重扭曲,但我的代码似乎没有任何内容可能导致这种情况。我的代码如下所示:
#vel_array, Rpl_array, and mass_loss_values are all some lists with the same length, but which I have not given here.
# Set up a regular grid of interpolation points
xi, yi = np.linspace(min(np.array(vel_array)), max(np.array(vel_array)), 100), np.linspace(min(np.array(Rpl_array)), max(np.array(Rpl_array)), 100)
xi, yi = np.meshgrid(xi, yi)
# Interpolate
rbf = Rbf(np.array(vel_array), np.array(Rpl_array), np.array(mass_loss_values), function='cubic')
zi = rbf(xi, yi)
mass_loss_values=np.array(mass_loss_values)
vel_array=np.array(vel_array)
Rpl_array=np.array(Rpl_array)
plt.imshow(zi, vmin=min(mass_loss_values), vmax=max(mass_loss_values), origin='lower',
extent=[min(vel_array), max(vel_array), min(Rpl_array), max(Rpl_array)])
plt.scatter(np.array(vel_array), np.array(Rpl_array), c=np.array(mass_loss_values))
plt.colorbar()
plt.show()
我会帮助你。
答案 0 :(得分:1)
x范围为40000,y范围仅为1000.
imshow
默认情况下将宽高比设置为equal
,从而产生非常大的宽高比(40:1)。尝试在aspect='auto'
命令中将imshow
设置为选项:
plt.imshow(
zi,
vmin=min(mass_loss_values),
vmax=max(mass_loss_values),
origin='lower',
extent=[min(vel_array), max(vel_array), min(Rpl_array), max(Rpl_array)],
aspect='auto'
)