python - Numpy:从字典中展平多个不同形状的数组

Question

我有一个包含不同形状的 np.array 的字典:

d = {
   "a" : np.array([1, 2, 3]), # shape (3,)
   "b" : np.array([4]), # shape (1,)
   "c" : np.array([[5, 6, 7], [8, 9, 10]]) # shape (2, 3)
}

现在我想将它们附加到一个向量中，如下所示:

output = np.array([1,2,3,4,5,6,7,8,9,10])

我该怎么做？

Answer 1

import numpy as np
from pandas.core.common import flatten

与 Pandas :

array=np.array(list(flatten(d.values())))

使用 numpy:

array=np.hstack([arr.ravel() for arr in d.values()])