将Python列表写入csv文件



我有一长串如下形式的列表---

a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]

即,列表中的值具有不同的类型——float、int、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.writer()的各种可选参数调整输出CSV的确切格式,如上面链接的库参考页中所述。

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]])

官方文件:http://docs.python.org/2/library/csv.html

在我的大列表中使用csv.writer花费了相当长的时间。我决定使用熊猫,它更快、更容易控制和理解:

 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")

如果出于任何原因想要手动执行(不使用csvpandasnumpy等模块(:

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("%sn" % ','.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])

在csv.writer函数中使用newline参数执行示例时,我收到了一条错误消息。以下代码对我有效。

 with open(strFileName, "w") as f:
    writer = csv.writer(f, delimiter=',',  quoting=csv.QUOTE_MINIMAL)
    writer.writerows(result)

如果将列表的列表lll导出到.csv,这将在Python3:中工作

import csv
with open("output.csv", "w") as f:
    writer = csv.writer(f)
    for element in lll:
        writer.writerows(element)

在这个页面上甚至没有看到一个答案,包括如何包含头以及如何创建文件。这里有一个方法也包含了这一点。方法适用于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]}) 

相关内容

  • 没有找到相关文章

最新更新