在Python中操作CSV文件通常使用内置的`csv`模块。下面是一些基本的操作步骤和示例代码:
读取CSV文件
import csv
打开CSV文件
with open('example.csv', newline='', encoding='utf-8') as csvfile:
创建csv阅读器对象
reader = csv.reader(csvfile)
遍历每一行数据
for row in reader:
print(row)
写入CSV文件
import csv
准备要写入的数据
data = [
['Name', 'Age'],
['Alice', 25],
['Bob', 30],
['Charlie', 35]
]
打开CSV文件以写入模式
with open('output.csv', 'w', newline='', encoding='utf-8') as csvfile:
创建csv写入器对象
writer = csv.writer(csvfile)
写入数据
writer.writerows(data)
使用`DictWriter`写入字典数据
import csv
准备要写入的字典数据
data = [
{'username': 'xian1', 'password': 'test1234'},
{'username': 'xian2', 'password': 'test1234'},
{'username': 'xian3', 'password': 'test1234'}
]
打开CSV文件以写入模式
with open('output.csv', 'w', newline='', encoding='utf-8') as csvfile:
创建csv写入器对象,指定字段名
fieldnames = ['username', 'password']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
写入表头
writer.writeheader()
写入数据
writer.writerows(data)
使用`pandas`库读取CSV文件
import pandas as pd
读取CSV文件
data = pd.read_csv('example.csv')
print(type(data))
print(data.head()) 打印前几行数据
使用`pandas`库写入CSV文件
import pandas as pd
准备数据
data = {
'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35]
}
创建DataFrame
df = pd.DataFrame(data)
写入CSV文件
df.to_csv('output.csv', index=False)
以上示例展示了如何使用Python内置的`csv`模块以及`pandas`库进行CSV文件的读取和写入操作。请根据您的具体需求选择合适的方法