Numpy MaskedArray.swapaxes()函数

numpy.ma.MaskedArray.swapaxes()函数可以交换数组的两个轴，即将数组的行和列转换为列和行。

语法

numpy.ma.MaskedArray.swapaxes(axis1, axis2)

参数

• axis1: 整数，交换轴索引。
• axis2: 整数，交换轴索引。

返回值

返回一个 ndarray，其轴已在 {axis1, axis2} 交换。

示例

import numpy as np
x = np.ma.array([[0, 1], [2, 3]], mask=[[False, True], [False, False]])
print(x)

输出：

[[0 --]
 [2  3]]

y = x.swapaxes(0,1)
print(y)

输出：

[[0  2]
 [-- 3]]

这里通过 np.ma.array 创建了一个 2x2 大小的数组，并传递了一个 mask 值，该 mask 对第一行第二个元素执行了屏蔽。 然后使用 swapaxes() 函数交换了轴，使元素按列排列。