Создайте CSV файл со значениями из списка Python

Я пытаюсь создать CSV файл со значениями из списка Python. Когда я печатаю значения в списке, они все unicode (?), То есть они выглядят примерно так.

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

Если я повторяю значения в списке i.e. for v in mylist: print v, они выглядят как обычный текст.

И я могу поставить , между каждым с print ','.join(mylist)

И я могу вывести файл, т.е.

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

Но я хочу выводить в CSV и иметь разделители вокруг значений в списке, например.

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

Я не могу найти простой способ включить разделители в форматирование, например. Я пробовал через инструкцию join. Как я могу это сделать?

Ответ 1

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 на w (см. Этот ответ SO)

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

Ответ 2

Вот безопасная версия Alex Martelli's:

import csv

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

Ответ 3

Лучший вариант, который я нашел, это использовать savetxt из numpy module:

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)

Ответ 4

Для другого подхода вы можете использовать DataFrame в pandas: И он может легко выгружать данные в CSV, как в коде ниже:

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

Ответ 5

Использовать модуль 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"

Ответ 6

В этом случае вы можете использовать метод 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-модуль может позаботиться обо всем этом для вас, позволяя вам выбирать между различными параметрами кавычек (все поля, только поля с кавычки и разделители, только не числовые поля и т.д.), а также как управлять кэшами esacpe (двойные кавычки или экранированные строки). Если ваши значения просты, string.join, вероятно, будет в порядке, но если вам нужно управлять множеством случаев, используйте доступный модуль.

Ответ 7

Блокнот Jupyter

Допустим, ваш список A

Затем вы можете закодировать следующее объявление, которое будет у вас в виде файла CSV (только столбцы!)

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

Ответ 8

вы должны обязательно использовать модуль CSV, но, скорее всего, вам нужно написать unicode. Для тех, кому нужно написать unicode, это класс из примера страницы, который вы можете использовать в качестве модуля утилиты:

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

Вот еще одно решение, для которого не требуется модуль 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

Это решение звучит безумно, но работает гладко, как мёд

import csv

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

Файл записывается csvwriter, поэтому свойства csv поддерживаются, то есть разделяются запятыми. Разделитель помогает в основной части, каждый раз перемещая элементы списка на следующую строку.

Ответ 11

>>>import csv

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