Pandas 读写csv

Pandas 是一个强大的数据分析工具，可以方便地读取和处理各种数据格式。其中，读写CSV文件是Pandas中一个十分常见的操作。下面是Pandas读写CSV文件的完整攻略，包括读取CSV文件、写入CSV文件以及一些实例说明。

读取CSV文件

Pandas提供了read_csv()函数可以用于读取CSV文件。以下是该函数的部分参数及说明：

• filepath_or_buffer：文件路径或BufferedReader对象
• sep：字段分隔符，默认为逗号
• header：指定header的行，如果没有header则传入None
• index_col：设置索引列，默认为None
• encoding：编码方式，默认为utf-8

以下是一个简单的读取CSV文件的例子，我们假设有一个名为data.csv的CSV文件需要读取：

import pandas as pd

df = pd.read_csv('data.csv', sep=',', header=0, encoding='utf-8')
print(df.head())

该代码将打印出CSV文件的前几行。

写入CSV文件

与读取CSV文件相似，Pandas还提供了to_csv()函数来将DataFrame对象写入CSV文件。以下是该函数的部分参数及说明：

• path_or_buf：文件路径或文件流对象
• sep：字段分隔符，默认为逗号
• header：是否写入header，默认为True
• index：是否写入索引列，默认为True
• encoding：编码方式，默认为utf-8

以下是一个将DataFrame对象写入CSV文件的例子：

import pandas as pd

data = {'name':['Tom', 'Jack', 'Sarah'], 'age':[20, 25, 30], 'gender':['M', 'M', 'F']}
df = pd.DataFrame(data)
df.to_csv('data.csv', sep=',', header=True, index=False, encoding='utf-8')

该代码将生成一个名为data.csv的CSV文件，其中包含了DataFrame对象的数据。

实例说明

以下是一个更为完整的示例，演示如何读取、处理并写入CSV文件。

1. 读取数据

假设有一个名为data.csv的CSV文件，该文件包含了以下数据：

name,age,gender
Tom,20,M
Jack,25,M
Sarah,30,F

以下是使用read_csv()函数读取该CSV文件的代码：

import pandas as pd

df = pd.read_csv('data.csv', sep=',', header=0, encoding='utf-8')
print(df.head())

该代码将输出以下结果：

    name  age gender
0    Tom   20      M
1   Jack   25      M
2  Sarah   30      F

1. 处理数据

对于读取的数据，我们可以进行各种各样的处理。例如，我们希望在原始数据的基础上增加一列，该列存储每个人的身高信息。以下是如何实现该功能的代码：

import pandas as pd

df = pd.read_csv('data.csv', sep=',', header=0, encoding='utf-8')
heights = [175, 180, 165]
df['height'] = heights
print(df.head())

该代码将输出以下结果：

    name  age gender  height
0    Tom   20      M     175
1   Jack   25      M     180
2  Sarah   30      F     165

1. 写入数据

在对数据进行处理后，我们可以将处理结果写入新的CSV文件中。以下是将处理结果写入新文件的代码：

import pandas as pd

df = pd.read_csv('data.csv', sep=',', header=0, encoding='utf-8')
heights = [175, 180, 165]
df['height'] = heights
df.to_csv('new_data.csv', sep=',', header=True, index=False, encoding='utf-8')
print('Data saved to "new_data.csv"')

该代码将生成一个名为new_data.csv的新CSV文件，并将处理结果写入该文件中。

总的来说，Pandas提供了方便的函数来读取和写入CSV文件，并且提供了强大的数据处理能力。使用Pandas读写CSV文件可以大幅提高数据处理效率，同时也使得我们的数据处理工作更加便捷。