获取当前scatter用的cmap
时间: 2023-06-30 10:18:31 浏览: 78
使用matplotlib中scatter方法画散点图
在 Matplotlib 中,可以使用 `scatter` 函数来绘制散点图。如果在绘制散点图时指定了 `c` 参数来指定每个点的颜色,那么 `cmap` 参数也可以被指定来指定颜色映射。如果想要获取当前散点图使用的颜色映射,可以通过 `scatter` 函数对象的 `cmap` 属性来获得。
以下是一个示例代码,展示了如何获取散点图中使用的颜色映射:
```python
import numpy as np
import matplotlib.pyplot as plt
# 随机生成一些数据
x = np.random.rand(50)
y = np.random.rand(50)
size = np.random.randint(50, 200, size=50)
color = x + y # 根据数据的和来进行彩色编码
# 绘制散点图并指定颜色映射
fig, ax = plt.subplots()
scatter = ax.scatter(x, y, s=size, c=color, cmap='cool')
plt.colorbar(scatter)
# 获取颜色映射
cmap = scatter.cmap
print(cmap)
plt.show()
```
在上面的代码中,我们先绘制了一个散点图,并指定了颜色映射为 `'cool'`。然后,我们使用 `colorbar` 函数来添加颜色条。最后,我们通过 `scatter.cmap` 属性来获得当前散点图使用的颜色映射,并将其打印出来。
阅读全文