matplotlib.colormaps[name]`` or ``matplotlib.colormaps.get_cmap(obj)怎么使用
时间: 2023-11-21 19:13:43 浏览: 182
`matplotlib.colormaps[name]` 可以通过 colormap 的名称来获取 colormap 对象,例如:
```python
import matplotlib.pyplot as plt
# 获取名为 'viridis' 的 colormap 对象
viridis_cmap = plt.cm.viridis
```
`matplotlib.colormaps.get_cmap(obj)` 可以通过字符串或者 colormap 对象来获取 colormap 对象,例如:
```python
import matplotlib.pyplot as plt
# 通过字符串获取名为 'viridis' 的 colormap 对象
viridis_cmap = plt.cm.get_cmap('viridis')
# 通过其他 colormap 对象获取对应的 colormap 对象
other_cmap = plt.cm.get_cmap(viridis_cmap)
```
获取到 colormap 对象之后,你可以把它传入到 `imshow()` 函数、`colorbar()` 函数等中去使用。例如:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义一张图片
img = np.random.rand(10, 10)
# 获取名为 'viridis' 的 colormap 对象
viridis_cmap = plt.cm.viridis
# 显示图片
plt.imshow(img, cmap=viridis_cmap)
plt.colorbar()
plt.show()
```
阅读全文