基于给定位信息的Numpy阵列指标提取

Question

我的数据以‘32位无符号整数’的形式出现，如下所示：

myData = np.array([1073741824, 1073741877, 1073742657, 1073742709, 1073742723, 1073755137, 1073755189,1073755969],dtype=np.uint32)

我希望获得“myData”元素的索引，其中发生以下情况：

Bit No. 0–1
Bit Combination: 00

我该怎么做呢？

Answer 1

您可以使用np.binary_repr()获取表示每个元素的字符串，并从该字符串获取与您的条件匹配的元素：

end = `00`
s = [np.binary_repr(ai, width=len(end)) for ai in myData]
indices = [i for (i, si) in enumerate(s) if si.endswith(end)]

编辑:矢量化(并推荐)方法

在进一步研究之后，我发现您可以在处理数组的np.unpackbits()视图之后使用uint8：

myData = myData.view(np.uint32) # not needed if it is already np.uint32
tmp = np.unpackbits(myData.view(np.uint8)[::4][None, :], axis=0)
indices = np.where((tmp[-2, :] == 0) & (tmp[-1, :] == 0))[0]

请注意，切片是根据您正在比较的位来获取的：

• 前8位的[::4]
• 第9位至第16位的[1::4]
• 第17至第24位的[2::4]
• 第25位至32位的[3::4]

例如，如果原始数据类型为np.uint64，则这个序列可能会继续下去，但在本例中，可以使用[n::8]。

Answer 2

我想说的是，典型的方法是：

np.flatnonzero((a & 0b11) == 0)

也就是说，屏蔽除后两位以外的所有位，并得到所有为零的索引。

这是短暂的，快速的，不作任何假设的endianness。