Python stringa CSV per l'array

Qualcuno sa di una semplice libreria o funzione per analizzare una stringa codificata CSV e trasformarla in un array o in un dizionario?

Non credo di volere il modulo csv integrato perché in tutti gli esempi che ho visto prende percorsi di file, non stringhe.

Puoi convertire una stringa in un oggetto file usando io.StringIOe quindi passarla al csvmodulo:

from io import StringIO
import csv

scsv = """text,with,Polish,non-Latin,letters
1,2,3,4,5,6
a,b,c,d,e,f
gęś,zółty,wąż,idzie,wąską,dróżką,
"""

f = StringIO(scsv)
reader = csv.reader(f, delimiter=',')
for row in reader:
    print('\t'.join(row))

versione più semplice con split()su newline:

reader = csv.reader(scsv.split('\n'), delimiter=',')
for row in reader:
    print('\t'.join(row))

Oppure puoi semplicemente split()trasformare questa stringa in righe usando \ncome separatore, e poi split()ogni riga in valori, ma in questo modo devi essere consapevole delle virgolette, quindi csvè preferibile usare il modulo.

Su Python 2 devi importare StringIOcome

from StringIO import StringIO

anziché.

Semplice: anche il modulo CSV funziona con le liste:

>>> a=["1,2,3","4,5,6"]  # or a = "1,2,3\n4,5,6".split('\n')
>>> import csv
>>> x = csv.reader(a)
>>> list(x)
[['1', '2', '3'], ['4', '5', '6']]

Il documento ufficiale per csv.reader() https://docs.python.org/2/library/csv.html è molto utile, che dice

oggetti file e oggetti elenco sono entrambi adatti

import csv

text = """1,2,3
a,b,c
d,e,f"""

lines = text.splitlines()
reader = csv.reader(lines, delimiter=',')
for row in reader:
    print('\t'.join(row))

>>> a = "1,2"
>>> a
'1,2'
>>> b = a.split(",")
>>> b
['1', '2']

Per analizzare un file CSV:

f = open(file.csv, "r")
lines = f.read().split("\n") # "\r\n" if needed

for line in lines:
    if line != "": # add other needed checks to skip titles
        cols = line.split(",")
        print cols

Come altri hanno già sottolineato, Python include un modulo per leggere e scrivere file CSV. Funziona abbastanza bene fintanto che i caratteri di input rimangono entro i limiti ASCII. Nel caso in cui si desideri elaborare altre codifiche, è necessario ulteriore lavoro.

La documentazione di Python per il modulo CSV implementa un'estensione di CSV.reader, che utilizza la stessa interfaccia ma può gestire altre codifiche e restituire stringhe unicode. Basta copiare e incollare il codice dalla documentazione. Successivamente, puoi elaborare un file CSV come questo:

with open("some.csv", "rb") as csvFile: 
    for row in UnicodeReader(csvFile, encoding="iso-8859-15"):
        print row

Per la documentazione:

E mentre il modulo non supporta direttamente le stringhe di analisi, può essere facilmente eseguito:

import csv
for row in csv.reader(['one,two,three']):
    print row

Trasforma la tua stringa in un unico elenco di elementi.

L'importazione di StringIO mi sembra un po 'eccessiva quando questo esempio è esplicitamente nei documenti.

https://docs.python.org/2/library/csv.html?highlight=csv#csv.reader

csvfile può essere qualsiasi oggetto che supporti il ​​protocollo iteratore e restituisca una stringa ogni volta che viene chiamato il suo metodo next ()

Quindi, a StringIO.StringIO(), str.splitlines()o anche un generatore sono tutti buoni.

Ecco una soluzione alternativa:

>>> import pyexcel as pe
>>> text="""1,2,3
... a,b,c
... d,e,f"""
>>> s = pe.load_from_memory('csv', text)
>>> s
Sheet Name: csv
+---+---+---+
| 1 | 2 | 3 |
+---+---+---+
| a | b | c |
+---+---+---+
| d | e | f |
+---+---+---+
>>> s.to_array()
[[u'1', u'2', u'3'], [u'a', u'b', u'c'], [u'd', u'e', u'f']]

Ecco la documentazione

Usalo per caricare un CSV in un elenco

import csv

csvfile = open(myfile, 'r')
reader = csv.reader(csvfile, delimiter='\t')
my_list = list(reader)
print my_list
>>>[['1st_line', '0'],
    ['2nd_line', '0']]

Panda è una libreria abbastanza potente e intelligente che legge CSV in Python

Un semplice esempio qui, ho file example.zip con quattro file al suo interno.

EXAMPLE.zip
 -- example1.csv
 -- example1.txt
 -- example2.csv
 -- example2.txt

from zipfile import ZipFile
import pandas as pd


filepath = 'EXAMPLE.zip'
file_prefix = filepath[:-4].lower()

zipfile = ZipFile(filepath)
target_file = ''.join([file_prefix, '/', file_prefix, 1 , '.csv'])

df = pd.read_csv(zipfile.open(target_file))

print(df.head()) # print first five row of csv
print(df[COL_NAME]) # fetch the col_name data

Una volta che hai i dati puoi manipolarli per giocare con un elenco o altri formati.