plt.scatter(x=data.age[data.target==1], y=data.chol[data.target==1], c="orange"
时间: 2024-02-22 18:58:32 浏览: 59
python matplotlib:plt.scatter() 大小和颜色参数详解
5星 · 资源好评率100%
这是一个使用matplotlib库绘制散点图的例子,其中`data`是一个数据集,`age`是数据集中的一个特征,`chol`是数据集中的另一个特征,`target`是数据集中的目标变量。
这个例子中,我们选择`target`变量为1的样本,以`age`作为x轴,`chol`作为y轴,绘制出散点图,并将散点的颜色设置为橙色。
下面是一个完整的例子:
```python
import matplotlib.pyplot as plt
import pandas as pd
# 读取数据集
data = pd.read_csv("heart.csv")
# 绘制散点图
plt.scatter(x=data.age[data.target==1], y=data.chol[data.target==1], c="orange")
plt.xlabel("age")
plt.ylabel("chol")
plt.show()
```
这个例子中,我们使用了pandas库来读取数据集,并使用matplotlib库来绘制散点图。
阅读全文