Pandas使用apply结合生成器列表报错：'generator' object is not callable

你的问题出在lambda表达式的写法以及返回值类型上，我们一步步拆解：

为什么会报错？ #

你写的代码：

df['category'] = df['title'].apply(lambda x: y for y in console_list if y in x)

这里的lambda x: y for y in console_list if y in x是无效语法——lambda函数的主体必须是单个表达式，而你直接把生成器表达式写在了冒号后面，Python会把它解析成错误的结构，最终返回了一个生成器对象。但pandas的apply期望每个元素处理后返回可序列化的值（比如字符串、None），不是生成器，所以抛出了TypeError: 'generator' object is not callable。

就算你修正语法把生成器用括号包起来：lambda x: (y for y in console_list if y in x)，也只是返回生成器对象，不是你需要的主机名称字符串，依然不符合需求。

正确的实现方式 #

根据你的需求（从标题中提取匹配的主机名称作为category列），有两种常用的高效实现方式：

方式1：修正lambda，用next()获取第一个匹配项 #

我们可以在lambda中用生成器配合next()，获取第一个匹配的主机名称，同时处理没有匹配的情况（比如返回None）。另外建议忽略大小写，避免因为标题里的小写主机名（比如ps4）匹配不到列表里的大写（PS4）：

console_list = ['PS4', 'Xbox One', 'PC', 'PS3', 'PS2', 'Xbox 360', '3DS']
df['category'] = df['title'].apply(
    lambda x: next((y for y in console_list if y.lower() in x.lower()), None)
)

• next(generator, default)：从生成器中取出第一个元素，如果没有匹配项，返回None（你也可以改成空字符串等自定义默认值）
• lower()：统一大小写，保证(ps4)能匹配到PS4

方式2：用pandas矢量化操作（更适合大型DataFrame） #

对于大型DataFrame，apply的效率较低，推荐用pandas的str.extract结合正则表达式，实现矢量化提取：

import re

console_list = ['PS4', 'Xbox One', 'PC', 'PS3', 'PS2', 'Xbox 360', '3DS']
# 构建正则表达式，转义特殊字符避免正则语法冲突
console_pattern = '|'.join(re.escape(c) for c in console_list)
# 提取匹配项，忽略大小写，最后转成大写统一格式
df['category'] = df['title'].str.extract(f'({console_pattern})', flags=re.IGNORECASE).str.upper()

这种方式比apply快很多，因为是pandas底层优化的矢量化操作，更适合处理大规模数据集。

测试验证 #

用你给出的示例数据测试：

import pandas as pd

data = {'title': ['Nier Automata (ps4)', 'Halo 5 Xbox One']}
df = pd.DataFrame(data)
# 用方式1或方式2处理后，得到的结果完全符合预期：
#                    title   category
# 0  Nier Automata (ps4)        PS4
# 1     Halo 5 Xbox One   Xbox One

内容的提问来源于stack exchange，提问作者Jasper