在二进制numpy数组中,我想选择一个,并将所有零转换为1。
例如:
np.array([0,0,1,0,0,0,0,1,0,0,0,0,0,0,1])
应该转换为:
np.array([0,0,1,1,1,1,1,1,1,1,1,1,1,1,1])
或者如果我有以下数组:
np.array([0,0,1,0,0,0,0,0,0,0,0,0,0,0,0])
np.array([0,0,0,0,0,0,0,0,0,0,0,0,0,0,0])
它们应保持不变。有没有一种快速的实施方法?
我的方法:
start, end = a.argmax(), a[::-1].argmax() + 1
# only update if there is at least a one:
if a[start]==1:
a[start:-end] = 1
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句