2017-01-22 105 views
1

我被困在一个问题。 我的词典列表看起来像这样:Python写一个词典列表csv

dict_list=[{'alfa':['001','2'], 'beta':['001','3'], 'gamma':['001','2']}, 
      {'alfa':['002','6'], 'beta':['002','4'], 'omega':['002','7']}] 

我的目的是创建这样一个CSV:

WORD, TEXT, FREQ 
alfa, 001, 2 
alfa, 002, 6 
beta, 001, 3 
beta, 002, 4 
gamma,001, 2 
omega,002, 7 

你认为是可能的解决呢?

+0

是的,继续尝试吧。然后,发布你的尝试,我们可以进一步帮助你。 –

+0

按内键(alpha,beta,...)分组是否需要,或者结果可以按任何顺序? – xzoert

+0

每本词典都有相同的长度吗? CSV中的行的顺序是否重要? – mikerose

回答

3

pandas提供了一种非常直观的方式来思考迭代字典列表。因为列表中的每个元素都可以很容易地转换为pandas.DataFrame,所以您可以遍历该列表,为每个元素创建一个DataFrame,然后将它们全部创建为concat

In [20]: l = [] 

In [21]: for dct in dict_list: 
    ...:  l.append( 
    ...:   pandas.DataFrame(dct).transpose() 
    ...: ) 

In [22]: tmp = pandas.concat(l) # aggregate them all 
In [23]: print(tmp) 
Out[23]: 
     0 1 
alfa 001 2 
beta 001 3 
gamma 001 2 
alfa 002 6 
beta 002 4 
omega 002 7 

pandas写入csv很容易,所以你可以这样做:

In [21]: tmp.to_csv('/my-file-path.csv') 

随着pandas对象就会有各种能力,然后对它们进行排序(如你想所期望的结果):

In [24]: tmp.sort_index() 
Out[24]: 
     0 1 
alfa 001 2 
alfa 002 6 
beta 001 3 
beta 002 4 
gamma 001 2 
omega 002 7 

以及更多!精确复制是你所寻找的,只是重命名Index和列,就像这样:

In [30]: tmp.index.name = 'WORD' 
In [36]: tmp = tmp.rename(columns={0: 'TEXT', 1: 'FREQ'}) 
In [37]: print(tmp) 

     TEXT FREQ 
WORD 
alfa 001 2 
beta 001 3 
gamma 001 2 
alfa 002 6 
beta 002 4 
omega 002 7 
+0

谢谢#benjamingross!它效果很好! – CosimoCD

+0

你打赌@CosimoCuriale,['pandas'](https://github.com/pandas-dev/pandas/blob/master/doc/cheatsheet/Pandas_Cheat_Sheet.pdf)是一个非常强大的模块,如果你打算做'csv'写作和数据操作,我强烈鼓励进一步的探索。 – benjaminmgross

2

创建从文件对象DictWriter对象,并转换为类型的字典的迭代后使用作家对象的writerows方法写入文件:如果您写入之前打印的迭代

import csv 

with open(your_filename, 'w') as f: 
    fieldnames = ['WORD', 'TEXT', 'FREQ'] 
    writer = csv.DictWriter(f, fieldnames=fieldnames) 
    writer.writeheader() 
    writer.writerows([dict(zip(fieldnames, (k,v1,v2))) for d in dict_list 
                 for k, [v1, v2] in d.items()]) 

文件,你必须:

# [{'FREQ': '2', 'TEXT': '001', 'WORD': 'alfa'}, 
# {'FREQ': '2', 'TEXT': '001', 'WORD': 'gamma'}, 
# {'FREQ': '3', 'TEXT': '001', 'WORD': 'beta'}, 
# {'FREQ': '6', 'TEXT': '002', 'WORD': 'alfa'}, 
# {'FREQ': '7', 'TEXT': '002', 'WORD': 'omega'}, 
# {'FREQ': '4', 'TEXT': '002', 'WORD': 'beta'}] 
+0

至少在Python 3上:* TypeError:'str'不支持缓冲区接口* – Tagc

+0

我仍然收到相同的错误。你是在Python 2还是Python 3下运行? – Tagc

+0

'open(your_filename,'w')'(no _b_)。 – DyZ

2

(总是)使用pandas

import pandas as pd 

df0 = pd.DataFrame(dict_list).stack().reset_index() 
# level_0 level_1   0 
#0  0 alfa [001, 2] 
#1  0 beta [001, 3] 
#2  0 gamma [001, 2] 
#3  1 alfa [002, 6] 
#4  1 beta [002, 4] 
#5  1 omega [002, 7] 

df0 = pd.concat([df0, df0[0].apply(pd.Series)], axis=1) 
df0.columns = ('dummy','WORD','tuple','TEXT','FREQ') 
df0[['WORD','TEXT','b']].sort_values('WORD').to_csv("your_file.csv",index=False) 
#WORD,TEXT,FREQ 
#alfa,001,2 
#alfa,002,6 
#beta,001,3 
#beta,002,4 
#gamma,001,2 
#omega,002,7 
1

这是我一分钱,如果我可以用pyexcel

>>> import pyexcel as p 
>>> dict_list=[{'alfa':['001','2'], 'beta':['001','3'], 'gamma':['001','2']}, 
...   {'alfa':['002','6'], 'beta':['002','4'], 'omega':['002','7']}] 
>>> s = p.Sheet() 
>>> for d in dict_list: 
...  s.row += p.get_sheet(adict=d, transpose_after=True) 
>>> s.colnames = ['WORD', 'TXT', 'FREQ'] 
>>> s 
pyexcel sheet: 
+-------+-----+------+ 
| WORD | TXT | FREQ | 
+=======+=====+======+ 
| alfa | 001 | 2 | 
+-------+-----+------+ 
| beta | 001 | 3 | 
+-------+-----+------+ 
| gamma | 001 | 2 | 
+-------+-----+------+ 
| alfa | 002 | 6 | 
+-------+-----+------+ 
| beta | 002 | 4 | 
+-------+-----+------+ 
| omega | 002 | 7 | 
+-------+-----+------+ 
>>> s.save_as('output.csv') 

的output.csv读起来像这个:

WORD,TXT,FREQ 
alfa,001,2 
beta,001,3 
gamma,001,2 
alfa,002,6 
beta,002,4 
omega,002,7