Question

我有一个名为“GO_file”的列表：

GO_file = ["A_1 12", "A_2 13", "A_3 14", "A_4 12", "B_1 1", "B_2 1", "B_3 5"]

我想将其转换为：

A：12,13,14

B：1,5

from collections import defaultdict
GO_file = ["A_1 12", "A_2 13", "A_3 14", "A_4 12", "B_1 1", "B_2 1" "B_3 5"]

GO_dict = defaultdict(list)
for GO_names in GO_file:
   gene_id = GO_names.split("_")[0]
   GO_id = GO_names.split(" ")[1:]
   GO_dict[gene_id] = GO_id
print GO_dict

但是，此代码仅附加键，只有一个值：

defaultdict(<type 'list'>, {'A': ['12'], 'B': ['5']})

我感谢任何建议。

Answer 1

您的代码几乎没有问题

您的GO_ID中有重复内容，您似乎只关心唯一身份。因此，您需要defaultdict(set)而不是defaultdict(list)
用于生成密钥和值的拆分算法是错误的
GO_dict[gene_id] = GO_id，只需将最后一个值分配给dict，而不是附加它。

可能的纠正解决方案

>>> GO_dict = defaultdict(set)
>>> for GO_names in GO_file:
   gene_id,_,GO_id = GO_names.partition(" ")
   gene_id = gene_id.split("_")[0]
   GO_dict[gene_id].add(GO_id)


>>> print GO_dict
defaultdict(<type 'set'>, {'A': set(['13', '12', '14']), 'B': set(['1', '5'])})

上述代码的一个可能问题是，不保证元素的顺序。不幸的是，默认库不提供OrderedSet，但我们可以轻松地自定义OrderedDict以服务我们的目的

>>> GO_dict = defaultdict(OrderedDict)
>>> for GO_names in GO_file:
   gene_id,_,GO_id = GO_names.partition(" ")
   gene_id = gene_id.split("_")[0]
   GO_dict[gene_id][GO_id] = None


>>> OrderedDict([('A', ['12', '13', '14']), ('B', ['1', '5'])])
OrderedDict([('A', ['12', '13', '14']), ('B', ['1', '5'])])

<强>但是

有些情况，我认为，itertools解决方案优于使用defaultdict

>>> from itertools import groupby
>>> from operator import itemgetter
>>> GO_file_kv = [(key.split("_")[0], value) 
                   for key, value in (elem.split(" ") for elem in GO_file)]
>>> {key: OrderedDict.fromkeys([e for _, e in value]).keys()
     for key, value in groupby(sorted(GO_file_kv, key=itemgetter(0)),
                       key=itemgetter(0))
 }
{'A': ['12', '13', '14'], 'B': ['1', '5']}

使用python defaultdict对列表中的项进行分组

1 个答案: