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