python - Pandas - 使用值列表创建较小的框架

Question

我有一个在大型 pandas 数据框中找到的值列表:

value_list = [1, 4, 5, 6, 54]

示例 DataFrame df 如下:

   column x
0  1      3
1  4      6
2  5      8
3  6      19
4  8      21
5  12     97
6  54     102

我想只使用这些值创建数据框的一个子集:

df_new = df[df['column'] is in value_list] # pseudo code

这可能吗？

Answer 1

您可能正在寻找 isin 操作。

In [60]: df[df['column'].isin(value_list)]
Out[60]:
   column    x
0       1    3
1       4    6
2       5    8
3       6   19
6      54  102

此外，您还可以使用查询，例如

In [63]: df.query('column in @value_list')
Out[63]:
   column    x
0       1    3
1       4    6
2       5    8
3       6   19
6      54  102