python - 搜索 pandas 数据框列，其输入值小于下一个索引列值

Question

我想搜索数据帧以查找与索引行和列值相对应的值。但是，我很挣扎，因为列标题值是范围的上限，其中前一个列标题值是下限(但其其他相邻列值的上限)。

我找不到将我的输入与范围下限对应的列相匹配的方法。

通过一个例子很容易看出:

data_frame_test = pd.DataFrame({'location' : [1, 2, 'S'],
                                '200' : [342, 690, 103],
                                '1000' : [322, 120, 193],
                                '2000' : [249, 990, 403]})

data_frame_test = data_frame_test.set_index('location')

我想做的是这个

location = 1
weight = 500

output = data_frame_test.iloc[1][??] #must be equal to 342

看，重量必须查看的列是 200，因为它在]200;1000[之间的范围内。我不知道还有什么可以尝试将其转换为 python 代码。任何帮助将不胜感激。

Answer 1

您可以创建一个自定义函数，它将迭代列以检查正确的列，然后返回该位置的单元格:

def get_val(location, weight, df):
    col = df.columns[0]
    for c in df.columns:
        if weight >= int(c):
            col = c
        else:
            break
    return df.loc[location, col]
get_val(1, 500, data_frame_test)