将 numpy ndarray 数组元素展平为特征行

Question

我有一个 4D numpy array ，其中每列代表 1 个数量，行是数量的统计导数，例如。

[mean mean mean
 std  std  std
 med  med  med]

假设第 1 列代表speed 2 列acceleration等。我想将所有可用数量的数组的每一列展平成一行特征形成：

mean std med mean std med mean std med ...

为了澄清我的担忧，我给出了以下MWE ：

input_shape = (1,3,4)
n_sample =20

X = np.random.randint(1, 10, size=(n_sample, )+ input_shape)
X.shape
(20,1,3,4)

所以 X 的前 2 个条目是：

X[:2]
array([[[[1, 7, 7, 8],
         [9, 3, 4, 1],
         [1, 1, 7, 1]]],


       [[[2, 8, 9, 4],
         [8, 4, 2, 7],
         [3, 9, 8, 4]]]])

列代表数量q1, q2, q3, q4 ，行代表统计数据：每个数组的mean, std, med 。

然后我想将每个数量的这些统计数据展平为特征行，以便最终结果是这样的：

  q1_mean q1_std q1_med q2_mean q2_std q2_med q3_mean q3_std q3_med q4_mean q4_std q4_med
     1      9       1      7      3      1      7       4       7      8      1       1
     2      8       3      8      4      9      9       2       8      4      7       4  

所以每个数组都是一个单一的观察。 所以我可以轻松转换为数据框，添加适当的列标题（名称）。

Answer 1

如果我理解正确，这应该做你想要的：

np.transpose(X,(0,1,3,2)).reshape(X.shape[0],-1)

请注意，转置轴 2 和 3 是必要的，因为重塑将以行​​主（C 样式）顺序完成。