插入几个新列，其值基于 pandas 中 Dataframe 中的另一列

Question

I have a dataframe of the variable shape. There are two fixed columns at the start plus flexible number of columns that will be different for every case. Starting dataframe:

A        B         FlexColumnA    FlexColumnB
Apples   Pears     0/1;23;45;67   1/1;23;45;67
Apples   Apples    0/0;24;26;27   0/1;27;28;29

I need to insert a new column before every "FlexColumn" with the following rules:

Name of a new column: "FlexColumn"+my_ending; my_ending is constant;

Values of a new column: "If a row of the flexible column contains '1/1', then 'norm'; if a row contains o/1, then insert 'half'; otherwise -'not_known' "

A         B        FlexColumnA_myEnding  FlexColumnA    FlexColumnB_myEnding FlexColumnB
Apples   Pears     half                  0/1;23;45;67   norm                 1/1;23;45;67
Apples   Apples    not_known             0/0;24;26;27   half                 0/1;27;28;29

Answer 1

How about something like this:

def get_outcome(s):
    if '1/1' in s:
        return 'norm'
    elif '0/1' in s:
        return 'half'
    else:
        return 'not_known'

flexcols = [c for c in df.columns if 'FlexColumn' in c]

new_cols = []

for col in flexcols:
    new_cols.append(df[col].apply(get_outcome).rename(col+'_myEnding'))
    new_cols.append(df[col])

pd.concat([df['A'],df['B']]+new_cols,axis=1)