如何在 Pandas 中根据另一列的值动态选取指定列的元素_网络技术

如何在 Pandas 中根据另一列的值动态选取指定列的元素

发布时间 - 2026-01-02 00:00:00 点击率：次

本文介绍一种高效、向量化的方法，利用 `pd.factorize` 和 numpy 索引，在大型 dataframe 中根据某列指定的列名，从源 dataframe 中按行提取对应列的值。

在实际数据分析中，常遇到“列名由另一列动态决定”的场景：例如，df1['idx'] 存储了要查询的列名（如 "a" 或 "b"），而真实数值存储在另一个结构对齐的 DataFrame df 中。目标是为每一行，依据 df1.idx 的值，从 df 的对应列中取出该行的值——且需兼顾性能，避免 .apply() 或 Python 循环。

以下是最优解法（纯向量化、无显式循环）：

import pandas as pd
import numpy as np

# 构造示例数据
df = pd.DataFrame({'a': [94, 170, 5],
                   'b': [31, 115, 8]}, index=[11, 12, 13])
df1 = pd.DataFrame({'idx': ["a", "b", "a"]}, index=[11, 12, 13])

# 核心步骤：向量化列查找
idx_codes, col_labels = pd.factorize(df1['idx'])  # 将列名映射为整数编码
# reindex 使 df 行索引与 df1 对齐，并仅保留所需列（去重后）
aligned_df = df.reindex(index=df1.index, columns=col_labels)
# 使用 NumPy 高级索引：每行取 idx_codes[i] 列的值
result = aligned_df.to_numpy()[np.arange(len(df1)), idx_codes]

print(result)  # [ 94 115   5]

将结果作为新列加入 df1：

df1['out'] = result
print(df1)
#     idx  out
# 11   a   94
# 12   b  115
# 13   a    5

✅ 为什么高效？

pd.factorize 时间复杂度接近 O(n)，远优于 map 或 apply；
reindex(columns=...) 是列级筛选，不触发行拷贝；
to_numpy() + NumPy 高级索引（[row_indices, col_indices]）是底层 C 实现，内存连续、零 Python 解释开销。

⚠️ 注意事项：

要求 df 与 df1 的索引完全对齐（或至少 df1.index 是 df.index 的子集），否则 reindex(index=...) 会引入 NaN；
df1['idx'] 中的列名必须全部存在于 df.columns 中，否则 reindex(columns=...) 会填充 NaN —— 建议提前校验：assert set(df1['idx']).issubset(df.columns)；
若存在缺失列名，可改用 df.lookup(df1.index, df1['idx'])（但注意：DataFrame.lookup 在较新 Pandas 版本中已被弃用，且对重复索引支持不佳，不推荐用于生产环境）。

该方法适用于百万级行数据，实测在 100 万行 × 数十列场景下耗时稳定在毫秒级，是处理“列名驱动索引”问题的工业级首选方案。

# python # 编码 # app # 为什么

相关栏目：【网站优化151355 】【网络推广146373 】【网络技术251813 】【 AI营销90571 】

上一篇：光荣发新年贺图《零》姐妹恭祝大家新年快乐

下一篇：如何在键盘 Tab 导航进入 div 时触发事件

相关栏目网站优化
网络推广
网络技术
AI营销

最新文章 Sublime怎么一键压缩JS代码 Su sublime如何在搜索中使用正则表达式 Sublime如何设置透明窗口效果 Su mysql如何设计商品表结构_mysql css属性背景图不显示怎么办_通过检查路如何使用Golang实现排序_Golan 农历闰月是怎么回事_为合回归年加一月调整塑造《刺客信条》艾吉奥传奇的编剧离开育碧 1英里等于多少公里 1mile和km的换 css grid布局中行和列是如何定义的 PS批量旋转和翻转图片，快速校正图片方向 C# Swagger UI自定义方法 C OPPO手机九宫格和全键盘怎么切换_OP Go语言如何实现用户登录注册_Golan 1节飞行速度多少公里每小时 1节是多少公纸嫁衣8千子树第五章庙门怎么开启庙门 Laravel 多行数据编辑表单中实现逐明日之后如何提升钓鱼等级明日之后提升钓支付宝怎样查年度账单_支付宝年度账单查看 C# 多线程UI更新Dispatcher

上一篇：光荣发新年贺图《零》姐妹恭祝大家新年快乐

下一篇：如何在键盘 Tab 导航进入 div 时触发事件