使用python编写一个对目录下名为IRIScsv的数据集2D可视化的代码
以下是使用Python编写的对目录下名为”IRIS.csv“的数据集2D可视化的代码示例:
import pandas as pd
import matplotlib.pyplot as plt
# 读取数据集
df = pd.read_csv("IRIS.csv")
# 绘制散点图
plt.scatter(df['sepal_length'], df['sepal_width'], c=df['species'].map({'setosa': 'red', 'versicolor': 'green', 'virginica': 'blue'}))
plt.xlabel('Sepal Length')
plt.ylabel('Sepal Width')
plt.title('IRIS Dataset')
plt.show()
解释:
-
首先,我们使用Pandas库中的read_csv函数读取名为“IRIS.csv”的数据集文件,并将其存储在一个名为df的数据帧中。
-
然后,我们使用matplotlib.pyplot库中的scatter函数绘制散点图。我们将数据帧df中的sepal_length列作为x轴,sepal_width列作为y轴,并使用species列的值作为点的颜色。
-
接下来,我们使用xlabel、ylabel和title函数为图表添加标签和标题。
-
最后,我们使用show函数显示图表。
注意:在代码中,我们假设数据集中包含三种不同的鸢尾花,即setosa、versicolor和virginica,并将它们分别对应为红、绿和蓝色。如果数据集中包含不同的鸢尾花种类或颜色,您需要根据实际情况进行更改。
原文地址: https://www.cveoy.top/t/topic/bL3k 著作权归作者所有。请勿转载和采集!