我需要计算 numpy uint64 变量中尾随零和前导零的数量,所以现在我这样做:
# n > 0
n = np.uint64(100)
s = np.binary_repr(n)
trail_zeros = len(s) - len(s.rstrip('0'))
lead_zeros = 64 - len(s)
有没有不使用字符串的更好方法? 优先考虑的是速度。谢谢!
最佳答案
对于 [0,2**63)
中的数字,我们可以使用一些算术运算来获取二进制格式的前导和尾随零,从而跳过字符串操作 -
def get_leading_trailing_zeros(n):
a = (2**np.arange(64) & n)
lead_zeros = 64-a.argmax()-1
if n==0:
trail_zeros = 1
else:
trail_zeros = (a==0).argmin()
return lead_zeros,trail_zeros
关于python - 在 numpy 中向前和向后位扫描,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55009983/