Pandas中根据分类id列表查询对应匹配name值的实现方法

方案1：字典映射法（性能最优，推荐） #

先把id和name的对应关系转为映射字典，再遍历目标id列表直接取值，时间复杂度最低，适合各种数据量场景：

# 构造id到name的映射字典
id_name_map = df.set_index('id')['name'].to_dict()

category_id_list = [4, 1, 1, 2]
# 完全按输入列表的顺序和重复度匹配结果
result = [id_name_map[cid] for cid in category_id_list]

输出结果：

['motorcycle', 'person', 'person', 'bicycle']

方案2：pandas原生reindex法 #

如果不想写列表推导，可直接用reindex方法按照输入的id列表重排索引，一行代码得到结果：

category_id_list = [4, 1, 1, 2]
result = df.set_index('id').reindex(category_id_list)['name'].tolist()

输出结果和方案1完全一致。

注意说明 #

普通isin()方法只会过滤出存在匹配的行，不会保留输入列表的重复项，也不会遵循输入列表的顺序，不适用该需求场景。上述两种方案均完全保留输入列表的顺序与重复项。

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