在Pandas Python中子集DF

Question

我有一個df ：

date          cusip   value
2012-12-20     XXXX     4.23
2012-12-20     YYYY     6.34
2012-12-20     ZZZZ     8.12
2012-12-21     XXXX     5.78
2012-12-21     YYYY     6.62
2012-12-21     ZZZZ     9.09

我想對僅選擇列表中存在的cusips子集進行cusips ：

cusList = ('XXXX', 'ZZZZ')

sub_df將是：

date          cusip   value
2012-12-20     XXXX     4.23
2012-12-20     ZZZZ     8.12
2012-12-21     XXXX     5.78
2012-12-21     ZZZZ     9.09

有什么建議嗎？ 謝謝。

Answer 1

您可以使用Series方法isin ：

In [1]: df = pd.read_csv(cusp.csv, sep='\s+')

In [2]: df.cusip.isin(['XXXX', 'ZZZZ'])
Out[2]: 
0     True
1    False
2     True
3     True
4    False
5     True
Name: cusip

In [3]: df[df.cusip.isin(['XXXX', 'ZZZZ'])]
Out[3]: 
         date cusip  value
0  2012-12-20  XXXX   4.23
2  2012-12-20  ZZZZ   8.12
3  2012-12-21  XXXX   5.78
5  2012-12-21  ZZZZ   9.09