在数组中对表示节点网络的数字进行Numpy排序

Question

假设我有一个数组，它表示一个节点网络，其中连接的节点被描述为“from nodes”和“to nodes”：

a = array([(1, 2), (2, 3), (3, 4), (4, 5), (2, 6), (6, 7), (7, 8), (2, 9),
       (9, 10), (10, 11), (2, 12), (12, 13), (13, 14), (13, 15), (14, 16)], 
      dtype=[('fnode', '<i4'), ('tnode', '<i4')])

a['fnode']
array([ 1,  2,  3,  4,  2,  6,  7,  2,  9, 10,  2, 12, 13, 13, 14])
a['tnode']
array([ 2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16])

如何将“to nodes”组合到列表中，使它们共享相同的“from node”？在

我想要这种格式：

^{pr2}$

编辑

为了清楚起见，我希望没有“To nodes”（例如节点8）的“from nodes”与一个空列表相关联。在

Answer 1

使用collections.defaultdict：

d = defaultdict(list)
map( lambda (k,v) : d[k].append(v), a)
print d
>> Out[40]: defaultdict(<type 'list'>, {1: [2], 2: [3, 6, 9, 12], 3: [4]
: [7], 7: [8], 9: [10], 10: [11], 12: [13], 13: [14, 15], 14: [16]})