python-3.x - dataframe.apply() 同时保留列名

Question

我想在数据框中的每一行上应用一个函数。我正在使用 apply() 但它传递了一个 Series 对象，因此没有传递列名称。有没有办法使用 .apply() 并保留列名？

Answer 1

如果您想明智地应用该函数:

对于数据框中的每一行，您希望将整行传递给一个函数，并且在您希望使用列名调用值的函数中，您可以这样做:

# df

    name value
0   AA  1
1   BB  2
2   CC  3

def  function_2(row):
    return row['value'] * 2

df['col_b'] = df.apply(lambda row: function_2(row), axis=1)

输出:

    name value col_b
0   AA  1   2
1   BB  2   4
2   CC  3   6

Example source Link