python – 根据列值将列表转换为元组字典

我正在尝试在Python中转换以下列表:

headers = [ 'Col1','Col2','Col3','Col4' ]
data    = [[  '1','A','D','X']
           [  '2','A','D','X']  
           [  '3','A','C','X']  
           [  '1','B','F','X']
           [  '2','B','F','X'] ]

到使用Col2作为唯一键的元组字典,所以输出如下所示:

{ 
  'A': [( '1','D','X'),('2','D','X'),('3','C','X') ]
  'B': [( '1','F','X'),('2','F','X') ]
}

到目前为止,我已设法将其转置并将其压缩到一个字典作为键,但我坚持到那里.

transpose_result = map(list, zip(*data))
data = dict(zip(headers, transpose_result))

你能帮我吗?
提前致谢.

解决方法:

迭代数据列表项,并使用行[1]作为键构建新字典,并为行(列表项)行[:1]行[2:]:

>>> data = [
...     ['1','A','D','X'],
...     ['2','A','D','X'],
...     ['3','A','C','X'],
...     ['1','B','F','X'],
...     ['2','B','F','X'],
... ]

>>> d = {}
>>> for row in data:
...     d.setdefault(row[1], []).append(row[:1] + row[2:])
... 
>>> d
{'B': [['1', 'F', 'X'], ['2', 'F', 'X']],
 'A': [['1', 'D', 'X'], ['2', 'D', 'X'], ['3', 'C', 'X']]}

当字典中没有键时,dict.setdefault用于使用空列表填充字典,因此可以安全地完成list.append调用.

如果你使用collections.defaultdict,它可能更简单:

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> for row in data:
...     d[row[1]].append(row[:1] + row[2:])
... 
>>> d
defaultdict(<type 'list'>, {
    'A': [['1', 'D', 'X'], ['2', 'D', 'X'], ['3', 'C', 'X']],
    'B': [['1', 'F', 'X'], ['2', 'F', 'X']]})
>>> dict(d)  # to convert it back to dict
{'A': [['1', 'D', 'X'], ['2', 'D', 'X'], ['3', 'C', 'X']],
 'B': [['1', 'F', 'X'], ['2', 'F', 'X']]}
上一篇:python – 列表和元组之间有什么区别?


下一篇:Python – 带有元组的Pandas数据帧