根据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。