当变量与重复的另一个列值匹配时，Pandas将获取多个列值

Question

我有一个数据框：

A   |  B
1     123
2     321
1     456
2     999
1     888
3     333
2     422

我有一个包含（1,2,3）的变量x所以当我将它与变量x为1的A列匹配时，它应该给我：

123 
456
888

我现在使用的代码是：

df.loc(df[A] == x, 'B').iloc[0]

这只给我123.

P.S：我想在列表中添加这些值。

Answer 1

df.loc[df['A'] == x, 'B']

或

df[df['A'] == x]['B']