对CSV表格数据进行预处理

Question

现在我有一个光谱的csv文件，该如何对它进行数据的处理？我想要wavenumber和光谱数据组成一个输入对应strawberry或者non-strawberry的标签，比如第一个 wavenumber 899.327，光谱数据 0.002805507， 它们对应的是non-strawberry。

Answer 1

按你给出的dataframe形式，假如存在df里：

def search(wavenumber,value):
  row=df[df[0]==wavenumber].iloc[0]
  result=row[row==value]
  if not result.empty:
    col=result.index.to_list()[0]
    return df.loc[0,col]

省略了一点数据校验的内容，也没考虑找到多个符合结果的情况。有需要自己加上就是。
如果数据是按浮点数存的，要注意精度问题。