按行特定列表选择熊猫数据框列

d_g*_*gnz 4 dataframe python-3.x pandas

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

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"]
Run Code Online (Sandbox Code Playgroud)

结果如下所示:

[1,2,11,6,5]
Run Code Online (Sandbox Code Playgroud)

ank*_*_91 7

只要lookup将罚款:

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