使用pandas生成/读取csv文件的方法实例

使用pandas生成/读取csv文件是很常见的操作。下面将详细介绍如何使用pandas生成/读取CSV文件。

生成CSV文件

生成CSV文件的方法很简单，可以先将数据存储在pandas的DataFrame中，然后使用DataFrame.to_csv方法将其保存为CSV文件。

步骤1：生成数据

在本示例中，我们将使用一个简单的学生信息数据集。首先，我们生成一个DataFrame，包含学生的姓名、年龄和性别数据。

import pandas as pd

students = {"Name": ["John", "Mary", "Peter"],
            "Age": [24, 21, 23],
            "Gender": ["Male", "Female", "Male"]}

df = pd.DataFrame(students) #将数据存储到DataFrame中

数据存储在DataFrame中后，我们可以使用DataFrame.head方法查看前几行数据：

print(df.head())

输出结果：

    Name  Age  Gender
0   John   24    Male
1   Mary   21  Female
2  Peter   23    Male

步骤2：生成CSV文件

将数据存储在DataFrame后，我们可以使用DataFrame.to_csv方法将其保存为CSV文件：

df.to_csv('students.csv', index=False)

to_csv方法有很多参数，本例中使用了index=False来去掉CSV文件中的行索引。

读取CSV文件

在pandas中读取CSV文件也很简单，可以使用pandas.read_csv方法读取CSV文件。下面我们将使用两个示例说明如何读取CSV文件。

示例1：读取简单的CSV文件

我们首先需要准备一个简单的CSV文件students.csv，它包含了三个学生的姓名、年龄和性别信息：

Name,Age,Gender
John,24,Male
Mary,21,Female
Peter,23,Male

我们可以使用pandas.read_csv方法读取它：

df = pd.read_csv('students.csv')

读取完毕后，我们可以使用DataFrame.head方法查看前几行数据：

print(df.head())

输出结果：

    Name  Age  Gender
0   John   24    Male
1   Mary   21  Female
2  Peter   23    Male

注意，我们没有指定任何read_csv的任何参数，因为CSV文件的格式很简单，读取CSV文件的默认参数就足够了。

示例2：读取带有UTF-8编码的CSV文件

如果CSV文件使用了UTF-8编码，我们需要指定encoding='utf-8'参数：

df = pd.read_csv('students2.csv', encoding='utf-8')

完整读取带有UTF-8编码的CSV文件的代码如下：

import pandas as pd

# 读取CSV文件
df = pd.read_csv('students2.csv', encoding='utf-8')

# 查看前几行数据
print(df.head())

上述代码会读取名为“students2.csv”的CSV文件，并将数据存储为名为“df”的DataFrame对象。然后它使用head()方法显示前五行数据。

以上便是使用pandas生成/读取csv文件的方法实例的完整攻略，希望能对你有所帮助！