numpy布尔过滤尺寸

时间:2019-07-01 17:30:49

标签: python numpy numpy-ndarray

我有一个3x3的numpy数组

arr = numpy.array([[1,2,3], [1,2,3], [1,2,3]])

我想从上面的数组中获取第一和第三行/列

filtered = arr[[True, False, True], [True, False, True]]

#this returns
numpy.array([1,3])

#however i was expecting
numpy.array([[1,3], [1,3]])

是否可以进行上述过滤?试图搜索问题,但找不到解决方法

0 个答案:

没有答案