如果值在列表中，则提取pandas子集

Question

我正在尝试提取DataFrame子集，我想要第一列在特定列表中的所有行。 例如，我有：

        t   gid
0   2010.0  1
1   2020.0  4
2   2040.0  6
3   2060.0  6
4   2090.0  12
5   2110.0  18
6   2140.0  1000

我想要一个gid在q_list中的所有行的DataFrame == [1,6,18]

Answer 1

您可以将列转换为索引并使用智能索引：

df.set_index('gid').ix[q_list]