我有一长串以下形式的列表---
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
即列表中的值具有不同的类型——浮点数、整数、字符串。如何将其写入 csv 文件,以便我的输出 csv 文件看起来像
1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2
我有一长串以下形式的列表---
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
即列表中的值具有不同的类型——浮点数、整数、字符串。如何将其写入 csv 文件,以便我的输出 csv 文件看起来像
1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2
Python 的内置CSV 模块可以轻松处理这个问题:
import csv
with open("output.csv", "wb") as f:
writer = csv.writer(f)
writer.writerows(a)
这假设您的列表定义为a
,因为它在您的问题中。您可以通过各种可选参数调整输出 CSV 的确切格式,csv.writer()
如上面链接的库参考页面中所述。
Python 3 的更新
import csv
with open("out.csv", "w", newline="") as f:
writer = csv.writer(f)
writer.writerows(a)
你可以使用pandas
:
In [1]: import pandas as pd
In [2]: a = [[1.2,'abc',3],[1.2,'werew',4],[1.4,'qew',2]]
In [3]: my_df = pd.DataFrame(a)
In [4]: my_df.to_csv('my_csv.csv', index=False, header=False)
import csv
with open(file_path, 'a') as outcsv:
#configure writer to write standard csv file
writer = csv.writer(outcsv, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL, lineterminator='\n')
writer.writerow(['number', 'text', 'number'])
for item in list:
#Write item to outcsv
writer.writerow([item[0], item[1], item[2]])
在我非常大的列表中使用 csv.writer 需要很长时间。我决定使用 pandas,它更快更容易控制和理解:
import pandas
yourlist = [[...],...,[...]]
pd = pandas.DataFrame(yourlist)
pd.to_csv("mylist.csv")
好的部分是你可以改变一些东西来制作一个更好的 csv 文件:
yourlist = [[...],...,[...]]
columns = ["abcd","bcde","cdef"] #a csv with 3 columns
index = [i[0] for i in yourlist] #first element of every list in yourlist
not_index_list = [i[1:] for i in yourlist]
pd = pandas.DataFrame(not_index_list, columns = columns, index = index)
#Now you have a csv with columns and index:
pd.to_csv("mylist.csv")
如果出于某种原因您想手动执行此操作(不使用csv
,pandas
等模块numpy
):
with open('myfile.csv','w') as f:
for sublist in mylist:
for item in sublist:
f.write(item + ',')
f.write('\n')
当然,滚动您自己的版本可能容易出错且效率低下……这通常就是为什么有一个模块的原因。但有时自己编写可以帮助您了解它们的工作原理,有时它会更容易。
如果您不想为此导入csv
模块,则可以仅使用 Python 内置函数将列表列表写入 csv 文件
with open("output.csv", "w") as f:
for row in a:
f.write("%s\n" % ','.join(str(col) for col in row))
Ambers 的解决方案也适用于 numpy 数组:
from pylab import *
import csv
array_=arange(0,10,1)
list_=[array_,array_*2,array_*3]
with open("output.csv", "wb") as f:
writer = csv.writer(f)
writer.writerows(list_)
将列表列表转储到 pickle 并使用pickle 模块恢复它怎么样?这很方便。
>>> import pickle
>>>
>>> mylist = [1, 'foo', 'bar', {1, 2, 3}, [ [1,4,2,6], [3,6,0,10]]]
>>> with open('mylist', 'wb') as f:
... pickle.dump(mylist, f)
>>> with open('mylist', 'rb') as f:
... mylist = pickle.load(f)
>>> mylist
[1, 'foo', 'bar', {1, 2, 3}, [[1, 4, 2, 6], [3, 6, 0, 10]]]
>>>
确保lineterinator='\n'
在创建作者时注明;否则,当数据源来自其他 csv 文件时,可能会在每个数据行之后将额外的空行写入文件...
这是我的解决方案:
with open('csvfile', 'a') as csvfile:
spamwriter = csv.writer(csvfile, delimiter=' ',quotechar='|', quoting=csv.QUOTE_MINIMAL, lineterminator='\n')
for i in range(0, len(data)):
spamwriter.writerow(data[i])
如果将lll
列表列表导出到.csv,这将在 Python3 中工作:
import csv
with open("output.csv", "w") as f:
writer = csv.writer(f)
for element in lll:
writer.writerows(element)
在 csv.writer 函数中使用换行符参数执行示例时,我收到一条错误消息。以下代码对我有用。
with open(strFileName, "w") as f:
writer = csv.writer(f, delimiter=',', quoting=csv.QUOTE_MINIMAL)
writer.writerows(result)
在此页面上甚至没有看到一个答案,其中包括如何包含标题以及创建文件。这是一种结合了它的方法。方法适用于 python 3
csv_filename = 'abc.csv'
fieldnames = ['Col_1','Col_2','Col_3','Col_4']
with open(csv_filename, mode='w') as csv_file:
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
writer.writeheader()
for i in my_list:
writer.writerow({fieldnames[0]: i[0], fieldnames[1]: i[1], fieldnames[2]: i[2],fieldnames[3]: i[3]})