python - 我正在处理一个银行数据集，我必须只使用 'experience in job' 列的整数部分

Question

我正在处理一个银行数据集，我必须只使用“工作经验”列的整数部分。我曾通过使用正则表达式定义一个函数并强制转换为整数来尝试此操作，但它不是整数而是存储地址。我做错了什么？

这是我的代码:

def year_converter(x):
    return int(re.search(r'\d+', x).group())    

bank_data['Years in current job'] = bank_data['Years in current job'].apply(lambda x: year_converter)

Before After applying my function

Answer 1

您作为 lambda 的结果传递了函数，这就是为什么它是您“之后”屏幕截图中的结果值。

执行lambda中的函数:

bank_data['Years in current job'] = bank_data['Years in current job'].apply(
    lambda x: year_converter(x))

或直接将其传递给 apply，以便 apply 正确执行它:

bank_data['Years in current job'] = bank_data['Years in current job'].apply(
    year_converter)