pandas使用参数应用函数

Question

我有一个带三个参数的函数。这是标题。

def count_ones(num, total_bits, group_size):

我正在尝试将此功能应用于数据列。但它没有回归我的预期。任何人都可以帮我解决这个问题吗？ total_bits为60，group_size为12。

df['events'] = df['data'].apply(count_ones, args =(60, 12))

Answer 1

将参数作为kwargs传递给apply：

df['events'] = df['data'].apply(count_ones, total_bits=60, group_size=12)

Answer 2

使用lambda：

def do_on_col(x, argument1):
  return x+argument1

df[col] = df[col].apply(lambda x: do_on_col(x, argument1))