python怎么写入csv文件

python怎么写入csv文件

在Python中，使用pandas库可以很方便地写入CSV文件。例如，我们有两组列表a和b，可以创建一个DataFrame来存储这些数据。首先导入pandas库，然后定义两组列表，接着用这些列表创建一个字典，其中键为列名。使用DataFrame的to_csv方法，将DataFrame写入CSV文件，设置index=False以避免写入行名。代码如下：

import pandas as pd
a = [1,2,3]
b = [4,5,6]
dataframe = pd.DataFrame({'a_name':a,'b_name':b})
dataframe.to_csv("test.csv",index=False,sep='')

这将生成一个CSV文件，内容如下：

a_name b_name
0 1 4
1 2 5
2 3 6

除了pandas，还可以使用内置的csv模块来逐行写入CSV文件。首先导入csv模块，然后打开文件，使用csv.writer对象写入数据。代码示例如下：

import csv
with open("test.csv","w") as csvfile:
writer = csv.writer(csvfile)
# 写入列名
writer.writerow(["index","a_name","b_name"])
# 写入多行
writer.writerows([[0,1,3],[1,2,3],[2,3,4]])

这将生成一个CSV文件，内容如下：

index a_name b_name
0 1 3
1 2 3
2 3 4

读取CSV文件时，可以使用pandas的read_csv方法，或者使用csv模块的reader。例如，使用pandas方法：

import pandas as pd
data = pd.read_csv('test.csv')

使用csv模块方法：

import csv
with open("test.csv","r") as csvfile:
reader = csv.reader(csvfile)
# 不需要readlines
for line in reader:
print(line)2024-12-14