Numpy温度计编码

Question

我正在尝试使用 numpy 优化的内置函数来生成温度计编码。如果给定长度为 1，则温度计编码基本上会生成n量。例如在 8-length 中，3 将被编码为：

1 1 1 0 0 0 0 0

使用 numpy 基于整数输入生成该向量基本上是切片和设置 1。

stream[:num_ones] = 1

所以我的问题是一个向量作为输入，例如生成矩阵输出的最佳方法是什么：

[2 3 4 1]

作为输入应该产生：

[[1 1 0 0 0 0 0 0],
 [1 1 1 0 0 0 0 0],
 [1 1 1 1 0 0 0 0],
 [1 0 0 0 0 0 0 0]]

我当前的解决方案是迭代所需大小的零矩阵，并使用我上面写的切片方法将所需的元素数设置为 1。我有没有更快的方法来做到这一点？

Answer 1

我以前从未听说过“温度计编码”，但是当您意识到它与单热编码如此相似时，很明显您可以使用位移操作到达那里：

>>> a = np.array([2, 3, 4, 1], dtype=np.uint8)
>>> print(np.fliplr(np.unpackbits((1 << a) - 1).reshape(-1,8)))
[[1 1 0 0 0 0 0 0]
 [1 1 1 0 0 0 0 0]
 [1 1 1 1 0 0 0 0]
 [1 0 0 0 0 0 0 0]]

编辑：您可以通过在 8 列块中工作来将想法推广到任意大小的整数：

a = np.array([2, 13, 4, 0, 1, 17], dtype=np.uint8)
out = np.empty((len(a), 0), dtype=np.uint8)
while a.any():
    block = np.fliplr(np.unpackbits((1 << a) - 1).reshape(-1,8))
    out = np.concatenate([out, block], axis=1)
    a = np.where(a<8, 0, a-8)

print(out)
[[1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]
 [1 1 1 1 1 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0 0 0 0 0]
 [1 1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]
 [1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]
 [1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0]]