在下面的代码中
import numpy as np
np.array([
[False, True, False],
[False, False, False],
[False, False, True],
[False, False, False]
])
我想获取与[True, False, True, False]
包含至少一个True的列表相对应的数组。
如评论中所建议,实现此目标的最佳方法是使用np.any,它测试沿给定轴的任何元素的求值是否为True。在实践中:
import numpy as np
a = np.array([
[False, True, False],
[False, False, False],
[False, False, True],
[False, False, False]])
np.any(a, axis=1)
array([ True, False, True, False])
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句