python – 如果列中的值位于值的设置列表中,则过滤数据帧行

参见英文答案 > How to implement ‘in’ and ‘not in’ for Pandas dataframe                                    7个
我有一个Python pandas DataFrame rpt：
@H_404_3@

@H_404_3@

rpt
<class 'pandas.core.frame.DataFrame'>
MultiIndex: 47518 entries, ('000002', '20120331') to ('603366', '20091231')
Data columns:
STK_ID                    47518  non-null values
STK_Name                  47518  non-null values
RPT_Date                  47518  non-null values
sales                     47518  non-null values

我可以像这样过滤库存ID为’600809’的行：rpt [rpt [‘STK_ID’] ==’600809′]@H_404_3@

@H_404_3@

<class 'pandas.core.frame.DataFrame'>
MultiIndex: 25 entries, ('600809', '20120331') to ('600809', '20060331')
Data columns:
STK_ID                    25  non-null values
STK_Name                  25  non-null values
RPT_Date                  25  non-null values
sales                     25  non-null values

我希望将一些股票的所有行放在一起,例如[‘600809′,’600141′,’600329’].这意味着我想要这样的语法：@H_404_3@

@H_404_3@

stk_list = ['600809','600141','600329']

rst = rpt[rpt['STK_ID'] in stk_list] # this does not works in pandas 

由于大熊猫不接受上述命令,如何实现目标？@H_404_3@

解决方法:@H_404_3@

使用isin方法. RPT [RPT [ ‘STK_ID’].ISIN(stk_list)].@H_404_3@