python - 根据列选择 pandas 数据框中的数据

Question

假设我有一个 pandas 数据框 df1 和值数组 a1

df1
col1 |col2 
A    |123
B    |321
C    |2323
D    |3232

a1
[A,B,A,D]

我想返回一个包含 len(a1) 行的数据帧 df2。

df2
    col1 |col2 
    A    |123
    B    |321
    A    |123
    D    |3232

我一直在尝试使用 iloc, loc 但找不到合适的解决方案...

编辑:我尝试过使用

len(df1[df1['col1'].isin(a1)]) 

但我认为这个解决方案只返回唯一的行，并且不会像我需要的那样重复它们。

Answer 1

您需要先set_index，然后才能将loc与您的值列表一起使用:

a1 = ['A', 'B', 'A', 'D']
df.set_index('col1').loc[a1]

      col2
col1
A      123
B      321
A      123
D     3232