用python把list里的数据写入csv

用python把list里的数据写入csv

在Python中，将列表数据写入CSV文件是一种常见的需求。使用pandas库可以轻松实现这一目标。首先，导入pandas库，并定义两个列表a和b。接着，创建一个DataFrame对象，其中列名分别为'a_name'和'b_name'，对应的值分别为a和b。最后，利用DataFrame的to_csv方法将数据写入名为"test.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=',')

执行上述代码后，"test.csv"文件的内容将如下：

a_name,b_name

1,4

2,5

3,6

除了使用pandas，还可以利用csv模块直接操作CSV文件。这里展示一种逐行写入的方法：

首先，打开一个名为"test.csv"的文件，然后创建一个csv.writer对象。接着，写入列名和多行数据。示例代码如下：

import csv

with open("test.csv","w") as csvfile:

writer = csv.writer(csvfile)

#先写入columns_name

writer.writerow(["index","a_name","b_name"])

#写入多行用writerows

writer.writerows([[0,1,3],[1,2,3],[2,3,4]])

执行上述代码后，"test.csv"文件的内容将如下：

index,a_name,b_name

0,1,3

1,2,3

2,3,4

读取CSV文件同样可以使用csv模块。首先，打开文件并创建一个csv.reader对象，然后逐行读取数据。示例代码如下：

import csv

with open("test.csv","r") as csvfile:

reader = csv.reader(csvfile)

#这里不需要readlines

for line in reader:

print(line)

以上就是Python中将列表数据写入CSV文件的两种方法。2024-12-06