参数化列名称

2024-03-28 14:41:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试参数化列名。这是我的密码

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'

如何参数化列名?你知道吗


Tags: import密码pandasdf参数mathpointspd
1条回答
网友
1楼 · 发布于 2024-03-28 14:41:51

与常规索引一样,需要使用双括号。以下内容相当于df[['Subject']]

>>> df[[c1]]
  Subject
0    MATH
1    MATH
2    MATH
3    MATH
4     PSY
5     PSY
6     PSY
7     PSY

相关问题 更多 >