Question

我想创建动态数据帧，在其中可以根据组值连接传入的数据场。我可以通过对数据帧名称进行硬编码来实现。以下是我正在使用的代码段：

group_keys = ['BAU_DYNAMIC', 'ML_DYNAMIC', 'RANDOM']

group_df是具有analytical_group

数据的数组

print(group_df[0])
    INDIVIDUAL_ID Segment analytical_group
2    100          Seg1      BAU_DYNAMIC
5    200          Seg1      BAU_DYNAMIC
12   900          Seg1      BAU_DYNAMIC
18   901          Seg1      BAU_DYNAMIC
20   910          Seg1      BAU_DYNAMIC

BAU_DYNAMIC_DF = pd.DataFrame()
ML_DYNAMIC_DF = pd.DataFrame() 
RANDOM_DF = pd.DataFrame()
for k,value in enumerate(group_keys):      
    if value == 'BAU_DYNAMIC':
        BAU_DYNAMIC_DF = pd.concat([BAU_DYNAMIC_DF,group_df[k]])         
    elif value == 'ML_DYNAMIC':
        ML_DYNAMIC_DF = pd.concat([ML_DYNAMIC_DF,group_df[k]])
    else:
        RANDOM_DF = pd.concat([RANDOM_DF,group_df[k]])

我想做这样的事情

for k,value in enumerate(group_keys): 
    grp_name = 'df_'+ value            
    if gbl[grp_name] == {}:
       gbl[grp_name] = pd.DataFrame(group_df[k])            
    else:
       gbl[grp_name] = pd.concat([gbl[grp_name],group_df[k]])

我该怎么做？

用循环在熊猫中连接多个动态数据框

0 个答案: