gpt4 book ai didi

python - Pandas DataFrame 作为带有嵌入列表的查找表

转载 作者:行者123 更新时间:2023-12-04 09:00:48 25 4
gpt4 key购买 nike

我有一个具有以下结构的数据框:

     A          B
[1, 2, 3] [a, b, c]
[4, 5, 6] [d, e, f]
我想查询数据框,以便在输入 1 时,它应该返回 [a,b,c] .同样,查询 6应该返回 [d, e, f] .最易读的方法是什么?

最佳答案

使用 maploc

n = 1
df.loc[df.A.map(lambda x: n in x), 'B']

Out[209]:
0 [a, b, c]
Name: B, dtype: object

关于python - Pandas DataFrame 作为带有嵌入列表的查找表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63569602/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com