使用设置为索引值的列访问列值

我有以下名为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')  

如何访问这些列?

评论
如今Ac
如今Ac
df["v1_v2_sum"] = df["V1"] + df["V2"]

无论如何,避免使用df.apply和UDF,它们的性能很差,并且只有在没有选项的情况下才需要。

点赞
评论