Python Pandas - 选择等于的数据帧列

Question

这个SQL代码的Pandas等价物是什么？

Select id, fname, lname from table where id = 123

我知道这相当于SQL＆＃39;其中＆＃39;熊猫中的条款：

df[df['id']==123]

这会选择特定列：

df[['id','fname','lname']]

但我无法弄清楚如何将它们结合起来。我在网上看到的所有例子都选择了有条件的所有列。我想选择具有一个或多个条件的有限数量的列。

Answer 1

使用类似SQL的.query()方法：

df.query("id == 123")[['id','fname','lname']]

或

df[['id','fname','lname']].query("id == 123")

或更多＆＃34; Pandaic＆＃34;：

df.loc[df['id'] == 123, ['id','fname','lname']]