Python快速安排列表的技巧

时间:2015-09-10 22:08:08

标签: python django

一次。 我有这个:

[
 [{'location': u'Austria',      u'number': 33},
 {'location':  u'Albania',      u'number': 29},
 {'location':  u'Afghanistan',  u'number': 666}],

 [{'location': u'Austria',      u'items': 6},
 {'location':  u'Albania',      u'items':  1},
 {'location':  u'Afghanistan'}, u'items':  0],

 [{'location': u'Austria',      u'loss': 1.0},
 {'location':  u'Albania',      u'loss': 2.0},
 {'location':  u'Afghanistan',  u'loss': 6.6}]
]

我已尝试生成如下列表:

[
 [u'Austria',     33,  6, 1.00], 
 [u'Albania',     29,  1, 2.00], 
 [u'Afghanistan', 666, 0, 6.60]
]

我试过这个:

country = [i['location'] for i in data[0]]
number =  [i['number'] for i in data[0]]
items =   [i['items'] for i in data[1]]
loss =    [i['loss'] for i in data[2]]

然后我拉下这个:

for i in range(0,len(country)):
        l.append([name[i], number[i], items[i], loss[i]])

它的工作,但是: 1)它的丑陋(对我而言) 2)它不是“一刀切”'

在其他时间我可能只有:

[
 [{'location': u'Austria',      u'number': 33},
 {'location':  u'Albania',      u'number': 29},
 {'location':  u'Afghanistan',  u'number': 666}],

 [{'location': u'Austria',      u'loss': 1.0},
 {'location':  u'Albania',      u'loss': 2.0},
 {'location':  u'Afghanistan',  u'loss': 6.6}]
]

我的代码将失败(超出索引),其他时间我可能在列表中有更多列表等。 这怎么样?

2 个答案:

答案 0 :(得分:6)

首先,我将其转置,以便所有澳大利亚人都在一起:

data_t = zip(*data)

然后我通过拉出子字典中的第一个非'location'键来累积行(因为这似乎是你正在寻找的字段)。

rows = []
for country_dat in data_t:
    country = country_dat[0]['location']
    row = [country]
    rows.append(row)
    for dct in country_dat:
        key = next(k for k in dct if k != 'location')
        row.append(dct[key])

答案 1 :(得分:0)

这个怎么样?

from collections import OrderedDict

od=OrderedDict()
for dl in data:
  for d in dl:    
    loc = d['location']
    od.setdefault(loc,{}).update(**d)

for k in od: print od[k]

{u'loss': 1.0, 'location': u'Austria', u'number': 33}
{u'loss': 2.0, 'location': u'Albania', u'number': 29} 
{u'loss': 6.6, 'location': u'Afghanistan', u'number': 666}

它会将任何一组目录键累积到一个目录中,并保留位置的原始顺序。

如果您想要一个包含任何顺序的任何字段的列表列表,请从此处开始:

fields=['location', 'loss', 'number'] 
lofl = [[ oo[k][x] for x in fields] for k in oo ]

>>> lofl
[[u'Austria', 1.0, 33], [u'Albania', 2.0, 29], [u'Afghanistan', 6.6, 666]]