数据框架示例-
col1 | col2col3 | col4 | colfromvaluestobepicked | new col | 1 | 1 | 0 | 1 | "col1"> | 1 |
---|---|---|---|---|---|
0 | 0 | 1 | 1 | "col2"> | 0 |
使用DataFrame.melt
作为查找的替代:
df1 = df.melt('colfromvaluestobepicked', ignore_index=False)
df['new']=df1.loc[df1['colfromvaluestobepicked'].str.strip("'") == df1['variable'],'value']
try this:
df['new col'] = df.apply(lambda row: row[row['colfromvaluestobepicked']],axis =1)