我有以下名为df的数据框:
V1 V2
IDS
a 1 2
b 3 4
如果我打印出索引和列,则结果如下:
> print(df.index)
Index(['a','b'],dtype='object',name='IDS',length=2)
> print(df.columns)
Index(['V1','V2'],dtype='object',length=2)
我想对这两列(按行)执行计算,并将其添加到新列。我尝试了以下操作,但似乎无法按预期访问该列。
df['sum'] = df.apply(lambda row: row['V1'] + row['V2'], axis=1)
运行最后一行代码时出现以下错误:
KeyError: ('V1', 'occurred at index a')
如何访问这些列?
无论如何,避免使用df.apply和UDF,它们的性能很差,并且只有在没有选项的情况下才需要。