进行数据处理时,需要将list转换为CSV文件,具体有以下两种方法:
方法一:使用pandas模块的to_csv方法实现将二维的list转为csv,具体地,使用其DataFrame属性,代码段如下:
import pandas as pd
# data1为list类型,参数index为索引,column为列名
data2 = pd.DataFrame(data = data1,index = None,columns = name)
# PATH为导出文件的路径和文件名
data2.to_csv(PATH)
需要注意的是,DataFrame中的参数index是每一行的索引,可以设置为None(默认为0,1,2...),也可以自己定义。但是目前我还不清楚如何去除,所以我选择了方法二。
方法二:使用csv模块中的writerow方法,逐行写入,具体如下:
import csv
# path为输出路径和文件名,newline=''是为了不出现空行
csvFile = open(path, "w+",newline='')
# name为列名
name = ['sessionId','itemId']
try:
writer = csv.writer(csvFile)
writer.writerow(name)
# data为list类型
for i in range(len(data)):
writer.writerow(data[i])
finally:
csvFile.close()