python - Pandas read_csv並使用usecols過濾列

woff

我有一個csv文件，當我用pandas.read_csv過濾列並使用多個索引與usecols輸出。

1. import pandas as pd
   
2. from StringIO import StringIO
   
3. 
   
4. csv = r"""dummy,date,loc,x
   
5. bar,20090101,a,1
   
6. bar,20090102,a,3
   
7. bar,20090103,a,5
   
8. bar,20090101,b,1
   
9. bar,20090102,b,3
   
10. bar,20090103,b,5"""
    
11. 
    
12. df = pd.read_csv(StringIO(csv),
    
13.         header=0,
    
14.         index_col=["date", "loc"],
    
15.         usecols=["date", "loc", "x"],
    
16.         parse_dates=["date"])

複製代碼

輸出

1. date          loc   x
   
2. 2009-01-01 a    1
   
3. 2009-01-02 a    3
   
4. 2009-01-03 a    5
   
5. 2009-01-01 b    1
   
6. 2009-01-02 b    3
   
7. 2009-01-03 b    5

複製代碼

文章出處