将Python的列表嵌套写入CSV文件

Question

将Python的列表嵌套写入CSV文件

249

我有一个长长的列表，其中每个元素都是以下形式的列表 ---

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

也就是说，列表中的值是不同的类型——浮点数、整数和字符串。我该如何将它写入CSV文件中，以便我的输出CSV文件看起来像这样

i.e. 列表中的值是不同的类型——浮点数、整数和字符串。我该如何将它写入CSV文件中，以便我的输出CSV文件看起来像

1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2

- user1403483

13个回答

49

你可以使用 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)

- Akavall

6

如果内置库csv可以胜任的话，我认为不应该使用pandas。 - Cloud

11

我喜欢熊猫，因为它很有力量。 - dorbodwolf

40

"Pandas" 确实很强大，但我不会开迈凯轮去附近的小店购物。 - user4698348

1

注意：对于列表嵌套列表非常大的情况，连续的csv写操作可能会使代码显着变慢。在这种情况下，使用pandas会更明智。 - anjandash

这个回答比之前的更有用，因为我的列表更复杂，里面包含了嵌套的列表。 - peetasan

43

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

- Dmitry Zagorulkin

2

如果您能够附上一些评论来解释您的答案，那么我会给您点赞。 - Burhan Khalid

3

"writerow" 不接受多个参数。 - Amber

1

w.writerow("a", "b", "c") 回溯（最近的调用最先）：文件“<stdin>”，第1行，<module> 类型错误：writerow（）需要恰好一个参数（给出了3个）

- Amber

@Amber 你使用的 Python 版本是哪个？ - Dmitry Zagorulkin

4

抱歉，我错过了。 - Dmitry Zagorulkin

15

在我的大型列表中使用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")

- Dinidiniz

11

如果由于某种原因您想手动完成此操作（而不使用像csv、pandas、numpy等模块）：

with open('myfile.csv','w') as f:
    for sublist in mylist:
        for item in sublist:
            f.write(item + ',')
        f.write('\n')

当然，自己编写的版本可能存在错误和低效...这通常就是为什么会有相应的模块。但有时编写自己的版本可以帮助你理解它们的工作原理，而有时候编写自己的版本也更加容易。

- tegan

8

如果你不想为此导入 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))

- fabda01

6

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

- Semjon Mössinger

5

把列表嵌套的内容用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]]]
>>>

- Good Will

3

创建写入器时，请确保指定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])

- yashi wang

2

在csv.writer函数中使用newline参数时，按照示例操作遇到了错误提示。下面的代码对我来说是可行的。

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

- Jie

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Amber · Accepted Answer

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)