plt.scatter(x[:, 0], x[:, 1], c=y, edgecolors='k') ,怎么改变散点的颜色
时间: 2024-04-14 16:27:42 浏览: 151
python matplotlib:plt.scatter() 大小和颜色参数详解
5星 · 资源好评率100%
要改变散点的颜色,您可以使用`c`参数来指定颜色。`c`参数可以接受不同的输入形式来指定颜色,包括预定义的颜色名称、HTML颜色代码、RGB元组、以及使用数值数据来映射到颜色的Colormap对象。
在您的例子中,`c=y`将目标变量`y`用作颜色映射。假设`y`是一个数值数组,您可以使用Colormap对象来为每个点指定不同的颜色。以下是一个示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建散点图数据
x = np.random.randn(100, 2)
y = np.random.randint(0, 2, size=100)
# 绘制散点图
plt.scatter(x[:, 0], x[:, 1], c=y, edgecolors='k', cmap='coolwarm')
# 添加颜色条
plt.colorbar()
# 显示图形
plt.show()
```
在上述示例中,`c=y`将目标变量用作颜色映射。使用`cmap='coolwarm'`来选择颜色映射,这将为每个点分配不同的颜色。最后使用`plt.colorbar()`添加一个颜色条。
您可以根据需要选择适当的颜色映射和参数设置。希望这对您有所帮助!如果还有其他问题,请随时提问。
阅读全文