如何在numpy中部分展平一个立方体(或higger dimensional ndarray)？

Question

例如，我正在尝试在一个轴上展平numpy中的3d数组(即，在一个轴上减少，在另一个轴上展平)，如果我有

X = array(
    [[[ 0, 1, 2, 3],
      [ 4, 5, 6, 7],
      [ 8, 9,10,11]],
     [[12,13,14,15],
      [16,17,18,19],
      [20,21,22,23]]])

我想在下面找到将X转换为X的操作：

array([
   [ 0, 1, 2, 3,12,13,14,15],
   [ 4, 5, 6, 7,16,17,18,19],
   [ 8, 9,10,11,20,21,22,23]])

我发现在这种情况下，np.concatenate((X[0],X[1]), axis=1)给出了解决方案，但是我想要一种更通用、更有效的方法来对N维numpy数组执行此操作。

Answer 1

使用numpy.transpose

>>> X.transpose(1, 0, 2).ravel()
array([ 0,  1,  2,  3, 12, 13, 14, 15,  4,  5,  6,  7, 16, 17, 18, 19,  8,
        9, 10, 11, 20, 21, 22, 23])