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]]。

如何获得类似的效果？