plt.subplots().colorbar的参数
时间: 2023-09-03 09:07:06 浏览: 94
对matplotlib改变colorbar位置和方向的方法详解
5星 · 资源好评率100%
`plt.subplots()` 函数返回一个元组,其中第一个元素是画布对象,第二个元素是子图对象。可以使用子图对象中的 `colorbar()` 方法创建一个颜色条。
`colorbar()` 方法有一些可选参数,下面是一些常用的参数:
- `mappable`: 必需的参数,指定颜色条的来源,通常是一个可绘制的对象,如 `imshow()` 或 `contourf()`。
- `ax`: 颜色条的子图对象,如果没有指定,则默认为当前子图。
- `orientation`: 颜色条的方向,可以是 `'horizontal'` 或 `'vertical'`,默认是 `'vertical'`。
- `shrink`: 颜色条的长度或宽度的缩小倍数,可以是一个小于 1 的浮点数,用于调整颜色条的大小和主图之间的间距,默认值是 1。
- `aspect`: 颜色条的长宽比,可以是一个浮点数或者字符串 `'auto'`,默认值是 `'auto'`。
- `fraction`: 颜色条的长度或宽度占整个画布的比例,可以是一个小于 1 的浮点数,默认值是 0.15。
例如,下面的代码演示了如何在一个图形中绘制一个热图和一个水平颜色条:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成一个随机矩阵
data = np.random.rand(10, 10)
# 绘制热图
fig, ax = plt.subplots()
im = ax.imshow(data)
# 添加颜色条
cbar = fig.colorbar(im, ax=ax, orientation='horizontal', shrink=0.6)
plt.show()
```
阅读全文