我有一个像这样的熊猫数据框
Windows Linux Mac
0 True False False
1 False True False
2 False False True
我想像这样将这三列组合在一个列中
OS
0 Windows
1 Linux
2 Mac
我知道我可以写一个像这样的简单函数
def aggregate_os(row):
if row['Windows'] == True:
return 'Windows'
if row['Linux'] == True:
return 'Linux'
if row['Mac'] == True:
return 'Mac'
我可以这样称呼
df['OS'] = df.apply(aggregate_os, axis=1)
问题是我的数据集很大，这个解决方案太慢了。有没有更有效的方法来进行这种聚合？