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

时间:2010-01-18 05:34:09

标签: python csv xlrd

我正在尝试使用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声明。我怎么能这样做?

13 个答案:

答案 0 :(得分:197)

import csv

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

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

要使它与python 3.x一起使用wb替换wsee this SO answer

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

答案 1 :(得分:90)

以下是Alex Martelli的安全版本:

import csv

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

答案 2 :(得分:25)

我发现的最佳选择是使用numpy module中的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)

答案 3 :(得分:21)

对于其他方法,您可以在DataFrame中使用pandas: 它可以轻松地将数据转储到csv,就像下面的代码一样:

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

答案 4 :(得分:12)

使用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"

答案 5 :(得分:7)

在这种情况下,您可以使用string.join方法。

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

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

或者作为单行

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

但是,您可能遇到的问题是您的字符串已嵌入引号。如果是这种情况,您需要决定如何逃避它们。

CSV module可以为您处理所有这些,允许您在各种引用选项(所有字段,仅包含引号和分隔符的字段,仅非数字字段等)之间进行选择以及如何控制esacpe charecters(双引号或转义字符串)。如果您的值很简单,string.join可能会正常,但如果您需要管理大量边缘情况,请使用可用模块。

答案 6 :(得分:2)

创建并写入csv文件

下面的示例演示如何创建和写入一个csv文件。 要创建动态文件编写器,我们需要导入一个软件包 import csv ,然后需要创建一个带有文件引用的文件实例 例如:-,以open(“ D:\ sample.csv”,“ w”,newline =“”)作为文件编写器

在这里,如果该文件不存在上述文件目录,则python将在指定目录中创建一个相同的文件,并且“ w”表示写入,如果要读取文件,则将“ w”替换为“ r”或附加到现有文件,然后添加“ a”。 newline =“”表示每次创建行时都会删除一个多余的空行,因此为了消除空行,我们使用newline =“”,使用诸如 fields = [“ Names”之类的列表创建一些字段名(列名) ,“年龄”,“类”] ,然后应用于类似 writer = csv.DictWriter(file_writer,fieldnames = fields) 在这里使用Dictionary writer并分配列名,使用 writer.writeheader()将列名写入csv,并使用 writer.writerow({“ Names”:“ John” ,“ Age”:20,“ Class”:“ 12A”}),而写文件值必须使用字典方法传递,这里的键是列名,值是您各自的键值

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

答案 7 :(得分:2)

这是 Python 3.x 的有效复制粘贴示例,其中包含用于定义自己的定界符和引用char的选项。

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"

注意:

如果引用设置为csv.QUOTE_MINIMAL,则.writerow()将引用 仅当字段包含定界符或quotechar时。这是 默认情况。

如果引用设置为csv.QUOTE_ALL,则.writerow()将引用所有 字段。

如果引用设置为csv.QUOTE_NONNUMERIC,则.writerow()将引用 所有包含文本数据的字段,并将所有数字字段转换为 浮动数据类型。

如果引用设置为csv.QUOTE_NONE,则.writerow()将转义 定界符而不是引用它们。在这种情况下,您还必须 为escapechar可选参数提供一个值。

答案 8 :(得分:1)

你应该确定使用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)

答案 9 :(得分:1)

这是另一个不需要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])

答案 10 :(得分:1)

Jupyter笔记本

假设您的列表为A

然后,您可以将以下广告编码为csv文件(仅限列!)

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

答案 11 :(得分:1)

这个解决方案听起来很疯狂,但像蜂蜜一样顺畅

import csv

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

该文件是由csvwriter写入的,因此csv属性得以保持,即逗号分隔。 分隔符每次都会将列表项移到下一行,从而在主要部分起到了帮助作用。

答案 12 :(得分:0)

对于那些寻求不太复杂的解决方案的人。我实际上发现了另一种简化的解决方案,可以完成类似的工作:

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)

希望这也有帮助。