欢迎访问宙启技术站
智能推送

如何在Python中读写CSV文件?

发布时间:2023-06-24 14:05:48

CSV(Comma-Separated Values)是一种常用的文件格式,它将数据以逗号的方式分隔开,一般用于存储表格数据。Python中有多种方式可以读取和写入CSV文件,本文将介绍其中的两种方式:

1. 使用csv模块

CSV模块在Python的标准库中。它提供了一组方法和类,可以很方便地读取和写入CSV文件。下面是使用CSV模块读取CSV文件的示例代码:

import csv

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

解析代码:

- csv.reader()方法用于读取CSV文件的数据,并返回一个可迭代的reader对象。

- for循环用于遍历可迭代的reader对象。

- 打印每一行的数据。

下面是使用CSV模块写入CSV文件的示例代码:

import csv

data = [
    ['Name', 'Age', 'Gender'],
    ['John', '25', 'Male'],
    ['Jane', '30', 'Female'],
    ['Bob', '35', 'Male']
]

with open('data.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    for row in data:
        writer.writerow(row)

解析代码:

- 使用csv.writer()方法创建一个CSV写入对象。

- for循环用于遍历data列表中的每一行数据。

- writer.writerow()方法用来写入一行数据到CSV文件中。

2. 使用pandas模块

pandas是Python的一种数据分析库,提供了更好的API来读取和写入各种格式的文件,包括CSV文件。下面是使用pandas模块读取CSV文件的示例代码:

import pandas as pd

df = pd.read_csv('data.csv')
print(df)

解析代码:

- pd.read_csv()方法用于读取CSV文件,并返回一个DataFrame对象。

- DataFrame对象是pandas中的一种数据结构,类似于Excel表格。

- 打印DataFrame对象。

下面是使用pandas模块写入CSV文件的示例代码:

import pandas as pd

data = {
    'Name': ['John', 'Jane', 'Bob'],
    'Age': [25, 30, 35],
    'Gender': ['Male', 'Female', 'Male']
}

df = pd.DataFrame(data)
df.to_csv('data.csv', index=False)

解析代码:

- pd.DataFrame()方法用于从数据集构建一个DataFrame对象。

- DataFrame对象是pandas中的一种数据结构,类似于Excel表格。

- df.to_csv()方法用于将DataFrame对象写入CSV文件中。