import pandas as pd import matplotlib.pyplot as plt from sklearn.model_selection import train_test_split # 数据分析 df = pd.read_csv('iris.csv', skiprows=[0], names=['sepal_length', 'sepal_width', 'petal_length', 'petal_width', 'class']) print(df.info()) print(df.describe()) print(df.isnull().sum()) # 随机抽取数据 train_data, test_data = train_test_split(df, test_size=0.3) train_data.to_csv('train_data.csv', index=False) test_data.to_csv('test_data.csv', index=False) # 数据可视化 df.plot(x='class', y=['sepal_length', 'sepal_width', 'petal_length', 'petal_width'], kind='line', title='Line Plot', figsize=(10, 5)) df.hist(bins=10, figsize=(10, 5)) df.plot(x='petal_length', y='petal_width', kind='scatter', title='Scatter Plot', figsize=(10, 5)) plt.show()
时间: 2023-09-23 13:09:28 浏览: 241
pandas.read_csv参数详解(小结)
这段代码是使用 Python 中的 pandas, matplotlib 和 sklearn 库对鸢尾花数据集进行了一些数据分析和可视化。
首先,通过 pandas 库读取了一个名为 iris.csv 的数据文件,该文件包含了鸢尾花的一些观测数据。然后使用 info(), describe() 和 isnull() 方法对数据进行了一些基本的统计分析,包括了数据的维度、每个属性的均值、标准差、最小值、最大值、缺失值数量等等。
接下来,使用 sklearn 库中的 train_test_split() 方法将数据集划分为训练集和测试集,并将它们保存到 train_data.csv 和 test_data.csv 文件中。
最后,使用 matplotlib 库对数据进行了一些可视化,包括了折线图、直方图和散点图。这些可视化将不同属性之间的关系展示出来,方便我们更好地理解和分析数据。
阅读全文