将列表的Python列表写入一个csv文件
我有一个很长的名单,如下所示:
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
即列表中的值是不同的types – 浮动,整型,string。我如何写入一个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()
如上面链接的库参考页所述。
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]])
官方文档: http : //docs.python.org/2/library/csv.html
你可以使用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)
Ambers的解决scheme也适用于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_)
如果因为什么原因你想手动(不使用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.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")
创build作者时一定要注明lineterinator='\n'
; 否则,当数据源来自其他csv文件时,每个数据行之后可能会有多余的空行写入文件…
这是我的解决scheme:
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])