如何在csv文件中查找标头索引？

Question

可以说我有一个csv列表，如下所示：

 A   B    C

 aa  bb   cc

 dd  ee   ff

 gg  hh   ii

我想知道B属于哪一列？

换句话说，对于以B为标题的列，我可以使用什么命令给我2号？

Answer 1

您可以使用pandas作为获取列标题作为列表，然后使用列表索引从0开始获取位置。

import pandas as pd

df = pd.read_csv('path/to/csv.csv')

df.columns.to_list().index('B')

Answer 2

您可以使用python中的csv模块，必须使用reader函数生成一个reader对象。

开发了阅读器功能，可以读取文件的每一行并列出所有列。然后，您必须选择要用于变量数据的列。

import csv
with open('data.csv','r')as f:
  data = csv.reader(f)
  headers=data[0]
  position=headers.index('B')    $1 

注意： Python中的索引从0而不是1开始。