plt.scatter函数设置颜色

`plt.scatter()` 函数可以通过参数 `c` 或 `color` 来设置散点图中点的颜色，具体方法如下：

- 使用单一颜色：将 `c` 或 `color` 设置为一个字符串，表示所有点使用同一种颜色。例如：`plt.scatter(x, y, c='red')` 或 `plt.scatter(x, y, color='red')`。
- 使用不同颜色：将 `c` 或 `color` 设置为一个数组或序列，表示每个点使用不同的颜色。例如：`plt.scatter(x, y, c=[0, 1, 2, 3, 4])` 将根据每个点的值来确定颜色。
- 使用颜色映射：将 `c` 或 `color` 设置为一个数组或序列，并使用 `cmap` 参数来指定颜色映射。例如：`plt.scatter(x, y, c=[0, 1, 2, 3, 4], cmap='viridis')` 将使用 viridis 颜色映射来确定每个点的颜色。

下面是一个使用颜色映射的例子：

import matplotlib.pyplot as plt
import numpy as np

x = np.random.randn(100)
y = np.random.randn(100)
colors = np.random.rand(100)

plt.scatter(x, y, c=colors, cmap='viridis')
plt.colorbar()
plt.show()

这个例子中，`x` 和 `y` 是随机生成的数据，`colors` 是一个由随机数生成的数组。`plt.scatter()` 函数根据 `colors` 中每个值的大小来确定每个点的颜色，同时使用 `cmap='viridis'` 来指定使用 viridis 颜色映射。最后使用 `plt.colorbar()` 函数来显示颜色条。