下面是一个例子：
m = np.array([[1, 2, 3], [2, 4, 3], [1, 2, 1]])
我想有一个包含m行的数组，其中任何一列中至少出现一个1，因此：
m
1
np.array([[1, 2, 3], [1, 2, 1]])
ymdaylpp1#
使用any和布尔索引：
any
out = m[(m==1).any(axis=1)]
输出量：
array([[1, 2, 3], [1, 2, 1]])
中间体：
(m==1) array([[ True, False, False], [False, False, False], [ True, False, True]]) (m==1).any(axis=1) array([ True, False, True])
1条答案
按热度按时间ymdaylpp1#
使用
any和布尔索引：输出量：
中间体：