python - 整数的 numpy bool 数组表示

Question

生成整数的 numpy bool 数组表示的最简单方法是什么？例如，将 6 映射到 np.array([False, True, True], dtype=np.bool)。

Answer 1

如果n是整数，表达式(n & (1 << np.arange(int(floor(log(n, 2) + 1))))) > 0将创建一个表示位的 bool 数组，最低有效位在第一个位置。

例如，

In [224]: n = 5

In [225]: from math import floor, log

In [226]: n = 5

In [227]: (n & (1 << np.arange(int(floor(log(n, 2) + 1))))) > 0
Out[227]: array([ True, False,  True], dtype=bool)

In [228]: n = 8

In [229]: (n & (1 << np.arange(int(floor(log(n, 2) + 1))))) > 0
Out[229]: array([False, False, False,  True], dtype=bool)

In [230]: n = 514

In [231]: (n & (1 << np.arange(int(floor(log(n, 2) + 1))))) > 0
Out[231]: array([False,  True, False, False, False, False, False, False, False,  True], dtype=bool)