为 apply function 中处理的每一行创建一个计数器

Question

我想知道当您将 function 应用于 Pandas dataframe 时，是否可以为每一行创建一个计数器，如下所示：

data['column'] = data.apply(lambda x: function(x['url']), axis=1)

output 应该与此类似：

1
2
3
4
5
...

谢谢

Answer 1

如果您的数据框有以数字命名的行索引，如下所示：

>> df = pd.DataFrame(
    [
        {'var_1': 23, 'var_2': 25},
        {'var_1': 24, 'var_2': 26},
        {'var_1': 25, 'var_2': 27},
        {'var_1': 26, 'var_2': 28},
    ]
)

>> print(df)
   var_1  var_2
0     23     25
1     24     26
2     25     27
3     26     28

你可以使用这段代码：

>> df['cnt'] = df.apply(lambda row: row.name + 1, axis=1)
>> print(df)
   var_1  var_2  cnt
0     23     25    1
1     24     26    2
2     25     27    3
3     26     28    4

但是，另一个选项，如何达到结果（没有.apply方法）可以是这个：

>> df['cnt_1'] = range(1, len(df) + 1)
>> print(df)
   var_1  var_2  cnt  cnt_1
0     23     25    1      1
1     24     26    2      2
2     25     27    3      3
3     26     28    4      4

Answer 2

一种可能的解决方案是将x.name参数添加到您的 function 并将print用于apply-indicator ：

data = pd.DataFrame({'url':['a','b','c']})

def function(x, y):
    #return progress indicator by index values
    print (y)
    #sample function
    return x + '_'
0    
1
2

data['column'] = data.apply(lambda x: function(x['url'], x.name), axis=1)
print (data)
  url column
0   a     a_
1   b     b_
2   c     c_