[英]How to make numba(nopython=true) work with 1D numpy.ndarray input with unknown number of elements
[英]Or operation of multiple 1d numpy.ndarray
我有多个相同形状的bool
np.ndarray
,例如:
a1 = [T, F, F]
a2 = [F, F, T]
a3 = [F, T, F]
我怎样才能应用or
操作可能是这样的并得到结果[T, T, T]
?
res = Or([a1, a2, a3])
如果我理解正确,你想要这个:
a1 = [True, False, False]
a2 = [False, False, True]
a3 = [False, True, False]
res = np.vstack([a1,a2,a3]).any(axis=0)
或同样:
res = np.logical_or.reduce([a1,a2,a3])
(我希望 reduce 比堆栈更快或性能相似)。 还有更多方法可以实现这一目标。 我希望这些比其他更快。
output:
[ True True True]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.