从 (a, b, c) 到 (a, c, b)

Question

我有几个形状的 Numpy 3D arrays （ a ， b ， c ）。 a 、 b和c的值未知。 但是，我想以有效的方式将每个 arrays 重塑为（ a ， c ， b ）。

这是我正在做的事情：

for array in list_of_arrays:
    a, b, c = array.shape
    array = array.reshape(a, c, b)

有没有更有效的方法来做到这一点，可能在一行代码中？ 我可以使用-1索引方法对 arrays 进行整形/重新排序吗？

谢谢你。

Answer 1

import numpy as np

# Example array with shape (2, 4, 6)    
array = np.arange(48).reshape((2, 4, 6))

# Swap axis in the 1st and 2nd dimension and print out its shape     
np.swapaxis(array, 1, 2).shape

Output：

(2, 6, 4)

Answer 2

也许np.transpose ？ 它将所有维度交换到指定的顺序。

x = np.random.randint(0, 256, (100, 80, 3))

np.transpose(x, (1, 0, 2))

(80, 100, 3)