另一个表中某个范围内的熊猫查找值

时间:2019-06-26 23:28:29

标签: python pandas dataframe

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

enter image description here

示例代码

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

1 个答案:

答案 0 :(得分:1)

这里不是使用IntervalIndex,而是使用numpy板载

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)