基於Python中的標題提取特定列中的數據

Question

我有一個數據集（csv文件），其中包含許多觀測孔隨時間變化的水位。例如

Time REM_WELL1 REM_WELL2 DEEP_WELL_SYP SP_1 
1989.778  4.54   
1989.786    4.87 
1989.806     
1990.531  4.46 4.5 4.7 
1990.553     
1990.575     
1990.603 4.66 4.49 4.54 4.75 
1990.908 4.71  4.61 4.6 
1990.992 4.72 4.61 4.63 4.67 
1991.106 4.41 4.59 4.58 4.82 
1991.156 4.61 4.41 4.5 4.77 
1991.256 4.61 4.37 4.49 4.75 

我想提取數據的特定井的列（例如：SP100）通過在第一線尋找它的名字和它搶水位繪製一個水文。誰能幫我這個？

Answer 1

使用csv模塊：

import csv 

with open('well.csv', 'rb') as f: 
    reader = csv.DictReader(f, dialect='excel-tab') 
    rows = list(reader) 

sp100 = [x.values() for x in rows if 'sp100' in x] 

或者，如果你需要的數據使用優秀pandas library做更多的操作。