在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

使用^{}模块：

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]

或者，如果需要对数据进行更多操作，请使用优秀的^{} library。在