Python: scrittura del dizionario annidato in CSV

Question

Sto provando a scrivere un dizionario nidificato in un file .csv. Ecco un semplice esempio:

import csv 
import itertools 

fields = [ 'org', '2015', '2014', '2013' ] 
dw  = { 'orgname1': { '2015' : 2, '2014' : 1, '2013' : 1 }, 
      'orgname2': { '2015' : 1, '2014' : 2, '2013' : 3 }, 
      'orgname3': { '2015' : 1, '2014' : 3, '2013' : 1 } 
     } 

with open("test_output.csv", "wb") as f: 
    w = csv.writer(f) 
    years = dw.values()[0].keys() 
    for key in dw.keys(): 
     w.writerow([key, [dw[key][year] for year in years]]) 

Questo mi fa una tabella con due colonne: la prima contiene orgname; il secondo contiene [2, 1, 1] (o i valori corrispondenti dal sottot dizionario). Mi piacerebbe un tavolo con quattro colonne: uno per orgname e poi tre per gli elementi di elenco corrispondenti.

Answer 1

Cambio:

w.writerow([key, [dw[key][year] for year in years]]) 

A:

w.writerow([key] + [dw[key][year] for year in years]) 

In caso contrario, si tenta di scrivere qualcosa di simile [orgname1, [2, 1, 1]] al CSV, mentre vuoi dire [orgname1, 2, 1, 1].

Come indicato da Padraic, è possibile cambiare years = dw.values()[0].keys() a years = sorted(dw.values()[0].keys()) o years = fields[1:] per evitare comportamenti casuali.

Answer 2

Questo appare come un lavoro per DictWriter:

import csv 
import itertools 
import sys 

fields = [ 'org', '2015', '2014', '2013' ] 
dw  = { 'orgname1': { '2015' : 2, '2014' : 1, '2013' : 1 }, 
      'orgname2': { '2015' : 1, '2014' : 2, '2013' : 3 }, 
      'orgname3': { '2015' : 1, '2014' : 3, '2013' : 1 } 
     } 

w = csv.DictWriter(sys.stdout, fields) 
for key,val in sorted(dw.items()): 
    row = {'org': key} 
    row.update(val) 
    w.writerow(row) 

Answer 3

implementazione alternativo utilizzando DictWriter e con le intestazioni

import csv 
import itertools 

fields = [ 'org', '2015', '2014', '2013' ] 
dw  = { 'orgname1': { '2015' : 2, '2014' : 1, '2013' : 1 }, 
      'orgname2': { '2015' : 1, '2014' : 2, '2013' : 3 }, 
      'orgname3': { '2015' : 1, '2014' : 3, '2013' : 1 } 
     } 

with open("test_output.csv", "wb") as f: 
    w = csv.DictWriter(f, fields) 
    w.writeheader() 
    for k in dw: 
     w.writerow({field: dw[k].get(field) or k for field in fields}) 

uscita:

org,2015,2014,2013 
orgname1,2,1,1 
orgname3,1,3,1 
orgname2,1,2,3 

Answer 4

Utilizzando DictWriter non v'è alcuna necessità di smistamento i campi in anticipo, dal w.writerow() assicurerà l'ordine corretto. Ma ha senso per ordinare gli oggetti stessi.

Così mettendo insieme tutti i suggerimenti di cui sopra e raccogliendo il meglio di ciascuno, mi sarebbe venuto su con seguente codice:

import csv 
import itertools 

def mergedict(a,b): 
    a.update(b) 
    return a 

fields = [ 'org', '2015', '2014', '2013' ] 
dw  = { 'orgname1': { '2015' : 2, '2014' : 1, '2013' : 1 }, 
      'orgname2': { '2015' : 1, '2014' : 2, '2013' : 3 }, 
      'orgname3': { '2015' : 1, '2014' : 3, '2013' : 1 } 
     } 

with open("test_output.csv", "wb") as f: 
    w = csv.DictWriter(f, fields) 
    w.writeheader() 
    for k,d in sorted(dw.items()): 
     w.writerow(mergedict({'org': k},d)) 

ho aggiunto un piccolo mergedict() funzione che lo rende un uno di linea più in basso.