将pandas数据帧与sql表连接以实现高效的select状态

2024-04-25 21:31:28 发布

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

假设一个名为fruits的sql表包含以下数据

id    name   color
1   apple     red
2   apple   green
3  orange  orange
4  banana  yellow

在python中,可以通过

^{pr2}$

现在我需要一个方法get_fruit_id(df),它根据df输入从表中返回id。因此,df

color   name
red     apple
green   apple
yellow  melon

我希望get_fruit_id(df)返回类似(1, 2, NaN)

color   name    id
red     apple   1
green   apple   2

一个可能的解决方案是从水果表中提取所有数据并使用熊猫。合并,但我要处理大表,所以我会优先考虑性能。提取所有数据对我来说似乎是不必要的。在

使用df的示例可以通过

df = pd.DataFrame({
    'name': ['apple']*2 + ['melon'],
    'color': ['red', 'green', 'yellow']
})

Tags: 数据nameidappledfsqlgetgreen

热门问题