Python:连接存储在字典中的数组

Question

我有一个大字典，其中存储了以下数组:

Store = dict()
Store['A'] = A #size 500x30
Store['B'] = B #size 500x20

我只有 A 和 B 用于说明。在我目前的现实生活中，我使用的字典中有大约 500 个键和值。

我想以优雅的方式连接数组以获得数组 C。

为了说明，这是我要实现的目标:

A = np.random.normal( 0, 1, ( 500, 20 ) )
B = np.random.normal( 0, 1, ( 500, 30 ) )
C = np.concatenate((A,B),1)

Answer 1

如果顺序无关紧要，将字典的值传递给 numpy.concatenate:

>>> store = {'A':np.array([1,2,3]), 'B':np.array([3,4,5])}
>>> np.concatenate(store.values(),1)
array([1, 2, 3, 3, 4, 5])

如果顺序很重要，你可以使用

np.concatenate([v for k,v in sorted(store.items(), key=...)], 1)

传入任何你喜欢的关键函数，或者如果你想按字典顺序排序，就把关键参数放在外面。可悲的是，concatenate 似乎没有采用生成器对象。