我有一个整数列表,以及一个不连续的二进制列表,我试图看看是否存在任何整数。
我可以使用numpy.digitize
来实现这种效果,然后简单地忽略任何偶数编号的垃圾箱,但我觉得必须采用不那么不优雅的方式来做到这一点。
例如,稍微简化数据,假设我有4个值; 3000000
,3275812
,4000000
和4657000
。我想知道这些是否属于两个箱子[3275812,3276886]
或[4656730,4657770]
。目前我有:
x = np.array([3000000, 3275812, 4000000, 4657000])
bins = np.array([3275812,3276886,4656730,4657770])
inds = np.digitize(x, bins)
inds
为我提供了值[0, 1, 2, 3]
,如果我忽视任何偶数的话,它会给我想要的结果。
有什么能明确地给我结果[0, 1, 0, 2]
或类似的东西而不必改变inds
吗?