ChatGPT解决这个技术问题 Extra ChatGPT

将 Python 列表写入 csv 文件

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

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

b
bucky

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。)
A
Akavall

您可以使用 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 确实很强大,但我不会用 McLaren 开车去隔壁的街角商店。
注意:对于列表非常大的情况,顺序 csv 写入操作会使代码显着变慢。在这种情况下,使用 pandas 会更明智。
M
Mike H-R
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


如果您可以用一些评论来解释您的答案,这将得到我的 +1。
writerow 不接受多个参数。
>>> w.writerow("a", "b", "c") Traceback(最近一次调用最后):文件“”,第 1 行,在 类型错误:writerow() 只接受一个参数(3)
@Amber 你用的是什么版本的python?
@Amber 对不起。我错过了 []
D
Dinidiniz

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

t
tegan

如果出于某种原因您想手动执行此操作(不使用 csvpandasnumpy 等模块):

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

当然,滚动您自己的版本可能容易出错且效率低下……这通常就是为什么有一个模块的原因。但有时自己编写可以帮助您了解它们的工作原理,有时它会更容易。


f
fabda01

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

S
Semjon Mössinger

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

G
Good Will

将列表列表转储到 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]]]
>>> 

J
Jonas Czech

创建 writer 时务必注明 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])

J
Jie

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

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

I
IAmBotmaker

如果将 listslistsllllist 导出到 .csv,这将在 Python3 中工作:

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

P
PanDe

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