使用numpy将数据放入不连续的容器

2024-04-27 03:58:42 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个整数列表,还有一个不连续的容器列表,我想看看是否有整数落入其中。你知道吗

我可以使用numpy.digitize来实现这个效果,然后简单地忽略任何偶数的箱子,但是我觉得一定有一种不那么不雅观的方法来实现这个效果。你知道吗

例如,稍微简化一下数据,假设我有4个值;3000000327581240000004657000。我想看看这些是不是落在两个箱子里,[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?你知道吗


Tags: 数据方法numpy列表np整数array容器