在numba中将整数转换为二进制数

Question

我正在使用 numba，并且想将整数转换为二进制表示。在普通的 python 中，我会使用 bin() 或 np.binary_repr()。但这些在 numba 中不受支持。有什么建议吗？我应该从头开始用 numba 编写一个转换器吗？

Answer 1

如果你仍然对这个 fn 通过 numba 生成 bin 数组感兴趣：

其中 size_of_bin 超出长度。

from numba import njit, prange
import numpy as np


@njit(fastmath=True)
def num_to_binary(number, size_of_bin = 32):
    out = np.zeros(size_of_bin)
    num = number
    index = 31

    for i in prange(size_of_bin):
        floatDivide = num // 2
        divide = num / 2
        if floatDivide != divide:
            out[index] = 1
    
        num = floatDivide
        
        index-= 1
        if index == -1 or floatDivide == 0:
            break

    return out