matplotlib contourf colorbar放到下方_Python空间绘图Colorbar详解
时间: 2024-01-18 11:04:58 浏览: 139
python可视化 matplotlib画图使用colorbar工具自定义颜色
5星 · 资源好评率100%
要将Matplotlib中的contourf图形的colorbar放在下方,可以使用subplot()函数创建一个包含两个子图的figure对象,一个子图用于绘制contourf图形,另一个子图用于绘制colorbar。
下面是一个示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成数据
x = np.linspace(-3, 3, 301)
y = np.linspace(-3, 3, 301)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 创建figure对象和两个子图
fig, (ax1, ax2) = plt.subplots(nrows=1, ncols=2, figsize=(8, 4))
# 绘制contourf图形
cf = ax1.contourf(X, Y, Z, cmap='coolwarm')
ax1.set_xlabel('X')
ax1.set_ylabel('Y')
ax1.set_title('Contourf Plot')
# 绘制colorbar
cbar = plt.colorbar(cf, ax=ax1, orientation='horizontal', shrink=0.8, pad=0.15)
cbar.ax.set_xlabel('Z')
cbar.ax.tick_params(labelsize=8)
# 在另一个子图中绘制一些其他的图形
ax2.plot(x, np.sin(x), label='sin(x)')
ax2.plot(x, np.cos(x), label='cos(x)')
ax2.set_xlabel('X')
ax2.set_ylabel('Y')
ax2.set_title('Other Plot')
ax2.legend()
plt.show()
```
在这个示例代码中,我们首先使用np.meshgrid()函数生成了一个二维的网格数据,然后使用contourf()函数绘制了一个contourf图形,并将其放在了第一个子图中。接着,我们使用colorbar()函数在第一个子图的右边绘制了一个水平方向的colorbar,并设置了一些参数,如缩放比例和位置等。最后,在第二个子图中我们绘制了一些其他的图形,如sin(x)和cos(x)。
使用这种方式可以方便地将contourf图形和colorbar放在同一个figure中,并且可以灵活地控制它们的位置和样式。
阅读全文