python-在numpy中进行正向和反向扫描

发布时间：2020-12-17 17:38:55 所属栏目：Python 来源：网络整理

导读：我需要计算numpy uint64变量中尾随和前导零的数量,所以现在我正在这样做： # n 0n = np.uint64(100)s = np.binary_repr(n)trail_zeros = len(s) - len(s.rstrip('0'))lead_zeros = 64 - len(s) 有没有使用字符串的更好的方法呢？优先是速度.谢谢！最佳答案

我需要计算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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!