要将最终绘制的热力图形状调整为正方形,可以使用ax1.set_aspect('equal')来设置轴对象的纵横比例。
以下是修改后的代码示例:
import numpy as np
import matplotlib.pyplot as plt
# 创建数据
Z = np.random.rand(10, 10)
# 创建新的图形和轴对象
fig, ax1 = plt.subplots(1)
heatmap = ax1.imshow(Z, origin="lower", cmap='hot')
# 添加颜色条轴
cbar = plt.colorbar(heatmap)
cbar.set_label('Fitness')
cbar.ax.yaxis.set_ticks_position('right')
ax1.set_title("ordinary kriging")
# 设置纵横比例以实现正方形效果
ax1.set_aspect('equal')
plt.tight_layout()
plt.show()
在这个示例中,我们使用ax1.set_aspect('equal')将轴对象的纵横比例设置为一致,从而使热力图呈现出正方形形状。
内容由零声教学AI助手提供,问题来源于学员提问




