gpt4 book ai didi

python-3.x - 按行特定列表选择 Pandas 数据框列

转载 作者:行者123 更新时间:2023-12-03 19:42:20 24 4
gpt4 key购买 nike

对于数据框中的每一行,我试图选择在列表中指定的列。该列表的长度与数据框的行长度相同。

df = pd.DataFrame({"a":[1,2,3,4,5], 
"b":[3,4,5,6,7],
"c":[9,10,11,12,13]})
lst = ["a","a","c","b","a"]

结果如下所示:
[1,2,11,6,5]

最佳答案

只是 lookup 会好的:

df.lookup(df.index,lst)
#array([ 1,  2, 11,  6,  5], dtype=int64)

关于python-3.x - 按行特定列表选择 Pandas 数据框列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61839231/

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