python - 检查行值是否等于列名并访问列的值

Question

示例数据框 -

col1	col2	col3	col4	colfromvaluestobepicked	新山
1	1	0	1	'col1'	1
0	0	1	1	'col2'	0

我想创建一个新列，其值基于 colfromvaluestobepicked ('col1') 是否为 col1，然后选择该 col 值并将其分配给新 col，依此类推。

我不确定如何实现这一目标？

标签： pythonpython-3.xpandas

Answer 1

试试这个：

df['new col'] = df.apply(lambda row: row[row['colfromvaluestobepicked']],axis =1)