用Python列表中的值创建一个.csv文件

我试图用Python列表中的值创建一个.csv文件。当我打印列表中的值时,它们都是unicode(?),即它们看起来像这样

[u'value 1', u'value 2', ...]

如果我遍历列表中的值,即for v in mylist: print v,它们似乎是纯文本。

我可以用print ','.join(mylist)在两者之间放一个,

我可以输出到一个文件,即。

myfile = open(...)
print >>myfile, ','.join(mylist)

但是我想输出到CSV,并在列表中的值周围有分隔符。

"value 1", "value 2", ...

我找不到一种简单的方法来在格式中包含分隔符,例如,我已经尝试通过join语句。我该怎么做呢?

840522 次浏览

你可以用绳子。方法。

为了清晰起见,分成几行——这是一个交互式会话

>>> a = ['a','b','c']
>>> first = '", "'.join(a)
>>> second = '"%s"' % first
>>> print second
"a", "b", "c"

或者是一条直线

>>> print ('"%s"') % '", "'.join(a)
"a", "b", "c"

然而,你可能有一个问题是你的字符串有嵌入引号。如果是这种情况,你需要决定如何逃离它们。

CSV模块可以为你处理所有这些,允许你在各种引用选项(所有字段,只有带引号和分隔符的字段,只有非数字字段,等等)和如何转义控制字符(双引号,或转义字符串)之间进行选择。如果你的值是简单的,字符串。join可能没问题,但如果你必须管理大量的边缘情况,使用可用的模块。

import csv


with open(..., 'wb') as myfile:
wr = csv.writer(myfile, quoting=csv.QUOTE_ALL)
wr.writerow(mylist)

编辑:这只适用于python 2.x。

让它与python 3一起工作。用w替换wb (看这个SO答案)

with open(..., 'w', newline='') as myfile:
wr = csv.writer(myfile, quoting=csv.QUOTE_ALL)
wr.writerow(mylist)

使用python的csv模块来读写逗号或制表符分隔的文件。csv模块是首选,因为它可以很好地控制引用。

例如,下面是你的工作示例:

import csv
data = ["value %d" % i for i in range(1,4)]


out = csv.writer(open("myfile.csv","w"), delimiter=',',quoting=csv.QUOTE_ALL)
out.writerow(data)

生产:

"value 1","value 2","value 3"

以下是Alex Martelli的安全版本:

import csv


with open('filename', 'wb') as myfile:
wr = csv.writer(myfile, quoting=csv.QUOTE_ALL)
wr.writerow(mylist)

你当然应该使用CSV模块,但你可能需要编写unicode。对于那些需要编写unicode的人,这是来自示例页面的类,您可以将其用作util模块:

import csv, codecs, cStringIO


class UTF8Recoder:
"""
Iterator that reads an encoded stream and reencodes the input to UTF-8
"""
def __init__(self, f, encoding):
self.reader = codecs.getreader(encoding)(f)


def __iter__(self):
return self


def next(self):
return self.reader.next().encode("utf-8")


class UnicodeReader:
"""
A CSV reader which will iterate over lines in the CSV file "f",
which is encoded in the given encoding.
"""


def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
f = UTF8Recoder(f, encoding)
self.reader = csv.reader(f, dialect=dialect, **kwds)


def next(self):
row = self.reader.next()
return [unicode(s, "utf-8") for s in row]


def __iter__(self):
return self


class UnicodeWriter:
"""
A CSV writer which will write rows to CSV file "f",
which is encoded in the given encoding.
"""


def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
# Redirect output to a queue
self.queue = cStringIO.StringIO()
self.writer = csv.writer(self.queue, dialect=dialect, **kwds)
self.stream = f
self.encoder = codecs.getincrementalencoder(encoding)()


def writerow(self, row):
self.writer.writerow([s.encode("utf-8") for s in row])
# Fetch UTF-8 output from the queue ...
data = self.queue.getvalue()
data = data.decode("utf-8")
# ... and reencode it into the target encoding
data = self.encoder.encode(data)
# write to the target stream
self.stream.write(data)
# empty queue
self.queue.truncate(0)


def writerows(self, rows):
for row in rows:
self.writerow(row)

我发现最好的选择是使用numpy模块中的savetxt:

import numpy as np
np.savetxt("file_name.csv", data1, delimiter=",", fmt='%s', header=header)

以防你有多个需要堆叠的列表

np.savetxt("file_name.csv", np.column_stack((data1, data2)), delimiter=",", fmt='%s', header=header)

下面是另一个不需要csv模块的解决方案。

print ', '.join(['"'+i+'"' for i in myList])

例子:

>>> myList = [u'value 1', u'value 2', u'value 3']
>>> print ', '.join(['"'+i+'"' for i in myList])
"value 1", "value 2", "value 3"

但是,如果初始列表包含一些“,则不会转义。如果需要,可以调用一个函数像这样转义它:

print ', '.join(['"'+myFunction(i)+'"' for i in myList])
对于另一种方法,你可以在熊猫中使用DataFrame: 它可以很容易地将数据转储到csv,就像下面的代码:

import pandas
df = pandas.DataFrame(data={"col1": list_1, "col2": list_2})
df.to_csv("./file.csv", sep=',',index=False)

Jupyter笔记本

假设你的列表名是A

然后您可以编写以下代码,并将其作为csv文件(仅限列!)

R="\n".join(A)
f = open('Columns.csv','w')
f.write(R)
f.close()

这个解决方案听起来很疯狂,但效果非常好

import csv


with open('filename', 'wb') as myfile:
wr = csv.writer(myfile, quoting=csv.QUOTE_ALL,delimiter='\n')
wr.writerow(mylist)
文件由csvwriter写入,因此csv属性被保留,即逗号分隔。 分隔符在主要部分的作用是每次将列表项移动到下一行

创建并写入csv文件

下面的例子演示了创建和写入csv文件。 为了创建一个动态文件写入器,我们需要导入一个包导入csv,然后需要创建一个带有file reference的文件实例 例:使用open("D:\sample.csv","w",newline="")作为文件写入器 < / p >

这里如果文件不存在于提到的文件目录中,那么python将在指定的目录中创建一个相同的文件,“w”表示写入,如果你想读取一个文件,那么将“w”替换为“r”,或者将“a”追加到现有文件中。newline=""指定它在每次创建行时删除一个额外的空行,因此为了消除空行,我们使用newline="",使用字段=[“名称”,“年龄”,“阶级”)这样的列表创建一些字段名(列名),然后应用于像这样的写入器实例 作家= csv.DictWriter (file_writer fieldnames =字段) 这里使用Dictionary writer并分配列名,要将列名写入csv,我们使用writer.writeheader (),要写入值,我们使用writer.writerow({“名称”:“约翰”,“年龄”:20,“类”:“12”}),而写入文件值必须使用Dictionary方法传递,这里的键是列名,值是各自的键值

import csv


with open("D:\\sample.csv","w",newline="") as file_writer:


fields=["Names","Age","Class"]


writer=csv.DictWriter(file_writer,fieldnames=fields)


writer.writeheader()


writer.writerow({"Names":"John","Age":21,"Class":"12A"})

下面是Python 3. x的复制粘贴示例,带有定义自己的分隔符和引号字符的选项。

import csv


mylist = ['value 1', 'value 2', 'value 3']


with open('employee_file.csv', mode='w') as employee_file:
employee_writer = csv.writer(employee_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_ALL)
employee_writer.writerow(mylist)

这将生成如下所示的employee_file.csv:

"value 1","value 2","value 3"

注意:

如果quote被设置为csv.QUOTE_MINIMAL,那么.writerow()将被引用 字段,如果它们包含分隔符或quotechar。这是

如果quote被设置为csv.QUOTE_ALL,那么.writerow()将引用所有 字段。< / p > 如果quote被设置为csv.QUOTE_NONNUMERIC,那么.writerow()将被引用 所有包含文本数据的字段,并将所有数值字段转换为

.浮点数据类型 如果quote设置为csv.QUOTE_NONE,则.writerow()将转义 分隔符,而不是引用它们。在这种情况下,你也必须这样做 为escapechar可选参数提供一个值

对于那些寻找简单解决方案的人。实际上,我发现了一个更简单的解决方案,可以完成类似的工作:

import pandas as pd
a = ['a','b','c']
df = pd.DataFrame({'a': a})
df= df.set_index('a').T
df.to_csv('list_a.csv', index=False)

希望这也能有所帮助。