python 画散点图对颜色设置
时间: 2023-09-12 19:12:05 浏览: 67
你可以通过 `c` 参数来设置散点图中每个点的颜色。`c` 参数可以接受一个数字列表或者字符串列表,列表中的每个元素都对应着散点图中一个点的颜色。下面是一个例子:
```python
import matplotlib.pyplot as plt
x = [1, 2, 3, 4]
y = [2, 4, 3, 1]
colors = ['r', 'g', 'b', 'y'] # 对应每个点的颜色
plt.scatter(x, y, c=colors)
plt.show()
```
在这个例子中,我们创建了一个包含四个点的散点图,每个点的横坐标和纵坐标分别为 `[1, 2, 3, 4]` 和 `[2, 4, 3, 1]`,并且我们通过 `colors` 列表来指定每个点的颜色。在 `plt.scatter()` 函数中,我们将 `c` 参数设置为 `colors`,这样就可以将每个点的颜色设置为对应列表中的颜色。
除了使用单个颜色列表来指定所有点的颜色,你还可以使用一个数字列表来指定每个点的颜色映射值。例如:
```python
import numpy as np
import matplotlib.pyplot as plt
x = np.random.rand(50)
y = np.random.rand(50)
colors = np.random.rand(50)
plt.scatter(x, y, c=colors)
plt.colorbar() # 添加颜色条
plt.show()
```
在这个例子中,我们通过 `np.random.rand(50)` 函数生成了包含 50 个随机数的列表 `x` 和 `y`,并且使用 `np.random.rand(50)` 函数生成另一个包含 50 个随机数的列表 `colors`。在 `plt.scatter()` 函数中,我们将 `c` 参数设置为 `colors`,这样就可以将每个点的颜色映射为对应列表中的值。同时,我们还添加了一个颜色条来显示颜色映射的范围。
阅读全文