python - Pandas:引用列名，不区分大小写

Question

使用 Pandas DataFrame，假设我在 csv 文件中有一堆列，我希望能够通过不区分大小写的名称访问其中的任何一个.

import pandas as pd

df = pd.read_csv(path_to_csv, delimiter=",")

df2 = df["Size"]

实际的列名是"Size"。我该怎么做才能使 df2 = df["sIZE"] 也被接受？

Answer 1

你可以调用str.lower在列上:

In [12]:
df = pd.DataFrame(columns=['Size','COLOUR','caTegory'])
df.columns

Out[12]:
Index(['Size', 'COLOUR', 'caTegory'], dtype='object')

In [14]:
df.columns = df.columns.str.lower()
df.columns

Out[14]:
Index(['size', 'colour', 'category'], dtype='object')