numpy - 2D Numpy 数组布尔切片

Question

我正在尝试将代码从 Matlab 移植到 Python，这涉及将 2D numpy 数组与另一个布尔数组切片。

z = np.array([[1,2,3],[4,5,6],[7,8,9]])
f = np.array([True,True,False])
print(z[f,f])

结果是一维数组 [1,5]

在 Matlab 中，这将返回 z 的左上角 2x2 部分，即 [[1,2],[4,5]]。

我怎样才能达到类似的效果？

标签： numpy