Question

我有一个数据框。我想说的是所有以“ A”或“ C”开头的列，然后将所有行除以列平均值。

我不明白为什么我的代码无法正常工作。

   dict_trans={'A':'STA'
    ,'B':'SUB'
    ,'C':'STA'}




for k, v in dict_trans.items():
    if df_train.columns.str.startswith(k):
        transformation=v
        if transformation='STA':
            df['STA_'+varname]=df[varname]/df.groupby(level=1)[varname].transform('mean')

Answer 1

请遵循以下代码。希望您尝试在以某些匹配字符开头并想要创建的某些列的新功能。

   # Creating a sample data frame : 
df=pd.DataFrame({'A_1':range(10),
         'A_2':range(10,20),
          'B_2':range(21,31),
         'C_1':range(32,42)})


columns_required=list(filter(lambda x:(x.startswith('A') or x.startswith('C') ),(df.columns).tolist()))

dict_trans={'A_1':'STA'
    ,'A_2':'SUB'
    ,'B_2':'STA',
     'C_1':'STA'}

for index,value in dict_trans.items():
    if index in columns_required and value=='STA':
        df['STA_'+index]=df.loc[:,index]/(df.loc[:,index]).mean()

结果是：

   A_1  A_2 B_2 C_1  STA_A_1        STA_C_1
0   0   10  21  32  0.000000    0.876712
1   1   11  22  33  0.222222    0.904110
2   2   12  23  34  0.444444    0.931507
3   3   13  24  35  0.666667    0.958904
4   4   14  25  36  0.888889    0.986301
5   5   15  26  37  1.111111    1.013699
6   6   16  27  38  1.333333    1.041096
7   7   17  28  39  1.555556    1.068493
8   8   18  29  40  1.777778    1.095890
9   9   19  30  41  2.000000    1.123288

Code snippet

遍历字典并应用数据帧转换

1 个答案:

结果是：