如何plt.scatter 画散点图 colorbar是分开的色阶
时间: 2023-07-09 11:26:22 浏览: 110
使用matplotlib中scatter方法画散点图
5星 · 资源好评率100%
要使用分开的色阶,需要先将数据分成几个部分,然后为每个部分指定一个颜色。可以使用matplotlib中的colormap来实现这个目的。以下是一个例子,假设有三个部分:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成一些随机数据
x = np.random.rand(100)
y = np.random.rand(100)
data = np.random.randint(0, 3, 100)
# 定义颜色映射
cmap = plt.cm.get_cmap('coolwarm', 3)
# 画散点图
plt.scatter(x, y, c=data, cmap=cmap)
plt.colorbar(ticks=[0, 1, 2])
plt.show()
```
在这个例子中,我们使用`np.random.randint`生成了0到2之间的随机整数,表示数据分成了三个部分。然后通过`plt.cm.get_cmap`获取一个名为`coolwarm`的颜色映射,并指定分成了3个部分。最后在`plt.scatter`函数中指定颜色为`data`,并将颜色映射设置为`cmap`即可。`plt.colorbar`函数可以添加一个颜色条,并将其分成三个部分。
阅读全文