I have a pandas dataframe like this:
aa bb cc dd ee
a a b b foo
a b a a foo
b a a a bar
b b b b bar
如果第1到4列的值为a
,我想要添加一个新的列。
结果将会是这样的:
aa bb cc dd ee ff
a a b b foo a
a b a a foo a
b a a a bar a
b b b b bar b
逻辑如下: 如果第1到4列中的任何一个值为
a
,则列ff
为a
,否则为b
。我可以定义一个函数,并手动处理每一列,例如:
def some_function(row);
if row['aa']=='a' or row['bb']=='a' or row['cc']=='a' or row[dd]=='a':
return 'a'
return 'b'
但我正在寻找一种可以跨越 n
个列进行扩展的解决方案。
感谢任何帮助!
df.iloc[:,:4].min(1)
? - Zero