首页 > 解决方案 > Pandas 在另一个表的范围内查找值

问题描述

在下面的示例中,我试图检查表 1 中的“值”是否在表 2中“开始”和“停止”列的值范围内。如果该值在该范围内,我想返回“水果”的类型。between 方法似乎能够完成此操作,但不确定如何将其应用于另一个表中的行。注意我需要使用大型数据集执行此任务,并且愿意使用 pandas 库之外的方法。

在此处输入图像描述

示例代码

df1 = pd.DataFrame({'Date': {0: '06-01', 1: '06-02', 2: '06-03', 3: '06-04'},
                   'Value': {0: 3, 1: 7, 2: 9, 3: 16}, })

df2 = pd.DataFrame({'Start': {0: 1, 1: 6, 2: 11, 3: 16},
                    'Stop': {0: 5, 1: 10, 2: 15, 3: 20},
                    'Fruit': {0: 'Apple', 1: 'Orange', 2: 'Pear', 3: 'Mango'},})

表格1

    Date  Value
0  06-01      3
1  06-02      7
2  06-03      9
3  06-04     16

表 2

    Fruit  Start  Stop
0   Apple      1     5
1  Orange      6    10
2    Pear     11    15
3   Mango     16    20

表 1 所需输出

    Date  Value  Fruit
0  06-01      3   Apple
1  06-02      7  Orange
2  06-03      9  Orange
3  06-04     16   Mango

标签: pythonpandasdataframe

解决方案


这是在路上,而不是使用IntervalIndex,我们与numpyboard-cast核对

s1=df2.Start.values
s2=df2.Stop.values
s=df1.Value.values[:,None]
np.dot((s>=s1)&(s<=s2),df2.Fruit)
array(['Apple', 'Orange', 'Orange', 'Mango'], dtype=object)

推荐阅读