参数化列名称

Question

我试图对列名称进行参数化。这是我的伪代码

import pandas as pd
df_scr_crd = {'Subject': ['MATH', 'MATH', 'MATH', 'MATH', 'PSY', 'PSY', 'PSY', 'PSY'], 
        'SCR_STRT': [10, 20, 30, 99999, 'A', 'B', 'C', 'D'], 
        'POINTS': [100, 200, 300, 500, 10, 20, 30, 40]}
df = pd.DataFrame(df_scr_crd, columns = ['Subject', 'SCR_STRT', 'POINTS'])

我有3个参数定义如下

c1 = 'Subject'
c2 = 'SCR_STRT'
c3 = 'POINTS'

我想通过变量调用数据框列名;例如，要致电df['Subject']，我想致电df[c1]，c1解析为'Subject'

如何参数化列名？

Answer 1

与常规索引一样，您需要使用双括号。以下内容相当于x：

df[['Subject']]