将 Python 列表写入 csv 文件

Posted 2023-02-23

【中文标题】将 Python 列表写入 csv 文件

【英文标题】：Writing a Python list of lists to a csv file

【发布时间】：2012-12-11 19:57:42

【问题描述】：

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

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

【参考方案1】：

Python 内置的CSV module 可以轻松处理这个问题：

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)

【讨论】：

为了兼容 Python 3，请从“wb”中删除“b”。

使用 Python 3 - open('output.csv', 'w', newline='')。如果我省略换行参数，我会得到一个额外的行。 docs.python.org/3/library/csv.html#csv.writer

在 python3 中我不得不使用 open('output.csv', 'w', newline="")

哇，python 3 错误非常无用。谢谢@vladV（需要一个类似字节的对象，而不是'str'）。事后看来，这有点道理，但根本不知道在哪里看。

@tlalco 这可能意味着您使用的是 Python 2，在这种情况下，您应该使用第一个代码块而不是第二个代码块。 （这也意味着您应该考虑切换到 Python 3。）

【参考方案2】：

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

【讨论】：

如果内置库csv可以做到，我认为不应该使用pandas。

我喜欢熊猫，因为它很强大

pandas 确实很强大，但我不会开迈凯轮车去隔壁的街角商店。

注意：对于列表非常大的情况，顺序 csv 写入操作会使代码显着变慢。在这种情况下，使用 pandas 会更明智。

【参考方案3】：

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

【讨论】：

如果你能用一些 cmets 解释你的答案，这将得到我的 +1。

writerow 不接受多个参数。

>>> w.writerow("a", "b", "c") Traceback（最近一次调用最后）：文件“”，第 1 行，在 TypeError : writerow() 只接受一个参数（给定 3 个）

@Amber 你用的是什么版本的python？

@Amber 对不起。我错过了[]

【参考方案4】：

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

【参考方案5】：

如果出于某种原因您想手动执行此操作（不使用 csv、pandas、numpy 等模块）：

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

当然，滚动您自己的版本可能容易出错且效率低下......这通常是为什么有一个模块来解决这个问题。但有时编写自己的代码可以帮助您了解它们的工作原理，有时甚至更容易。

【参考方案6】：

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

【参考方案7】：

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

【参考方案8】：

将列表列表转储到pickle并使用pickle module恢复它怎么样？这很方便。

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

【参考方案9】：

创建作者时一定要注明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])

【参考方案10】：

如果将 lists 的 lists 的 lll list 导出到 .csv，这将在 Python3 中工作：

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

【参考方案11】：

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

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

【参考方案12】：

在此页面上什至没有看到一个答案，其中包括如何包含标题以及创建文件。这是一种结合了它的方法。方法适用于 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])