python - 对数据帧的每一行应用函数并同时递增计数器

Question

我知道我可以使用 apply() 在数据帧的每一行上应用函数，如下所示:

import pandas as pd
df = pd.DataFrame({'Name' : ['A', 'B', 'C'], 'Number' : [1,2,3]})
def func(row):
    pass
df.apply(lambda x: func(x), axis =1 )

但我需要将增量计数器传递给 func()。类似于下面的内容，但我不知道如何增加计数器!

import pandas as pd
df = pd.DataFrame({'Name' : ['A', 'B', 'C'], 'Number' : [1,2,3]})
def func(row, counter):
    pass
counter = 0 #initial value
df.apply(lambda x: func(x, counter), axis =1 )

Answer 1

在Python中，functions are first class citizens ，即使是属于对象的，实现您想要的目的的一种方法如下:

import pandas as pd

df = pd.DataFrame({'Name': ['A', 'B', 'C'], 'Number': [1, 2, 3]})


class Counter:

    def __init__(self, seed):
        self.counter = seed

    def fun(self, n):
        if True:  # if you need to check a condition
            self.counter += 1  # add any value you see fit
        return n + self.counter


counter = Counter(0)

result = df["Number"].apply(counter.fun)
print(result)

输出

0    2
1    4
2    6
Name: Number, dtype: int64

请注意，一等公民身份意味着您不需要创建 lambda，您可以传递函数本身。