交换numpy数组的维数

Question

我要执行以下操作：

for i in dimension1:
  for j in dimension2:
    for k in dimension3:
      for l in dimension4:
        B[k,l,i,j] = A[i,j,k,l]

不使用循环。最后，A和B包含相同的信息，但索引不同。

我必须指出，尺寸1,2,3和4可以相同或不同。因此，numpy.reshape（）似乎很困难。

Answer 1

请注意：Jaime的答案更好。NumPynp.transpose正是为此目的而提供的。

在numpy中执行此操作的规范方法是使用np.transpose的可选置换参数。在您的情况下，从ijkl转到klij，排列是(2, 3, 0, 1)，例如：

In [16]: a = np.empty((2, 3, 4, 5))

In [17]: b = np.transpose(a, (2, 3, 0, 1))

In [18]: b.shape
Out[18]: (4, 5, 2, 3)

或者使用np.einsum
; 这可能是其预期目的的一种变形，但是语法非常不错：

In [195]: A = np.random.random((2,4,3,5))

In [196]: B = np.einsum('klij->ijkl', A)

In [197]: A.shape
Out[197]: (2, 4, 3, 5)

In [198]: B.shape
Out[198]: (3, 5, 2, 4)

In [199]: import itertools as IT    
In [200]: all(B[k,l,i,j] == A[i,j,k,l] for i,j,k,l in IT.product(*map(range, A.shape)))
Out[200]: True