子集的DF在熊貓的Python

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：

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 

任何r ecommendations？謝謝。

Answer 1

可以使用系列方法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