讲师中心 微信公众号
AI工具推荐 视频效率加速

如何对 Pandas DataFrame 的指定行索引高效应用自定义向量化函数

陌浩小哥_7076

陌浩小哥_7076

发布时间:2025-12-30 22:36:09

|

330人浏览过

|

来源于php中文网

原创

如何对 Pandas DataFrame 的指定行索引高效应用自定义向量化函数

本文介绍如何将返回多值的 python 函数(如 `computeleft`)仅应用于 dataframe 的特定行索引,并将结果精准填充至对应位置,其余位置自动设为 nan,兼顾可读性与性能。

在实际数据处理中,我们常需对 DataFrame 的部分行(而非全部)执行复杂计算,并将多维输出(如长度为 4 的数组)分别写入多个目标列。直接使用 np.vectorize 配合 signature="()->(4)" 虽能向量化,但默认作用于全量索引;若仅需处理子集(如 [2, 5, 7, 8, 10]),关键在于分离“初始化”与“选择性赋值”两个步骤。

首先,确保函数定义正确且支持向量化调用:

import numpy as np
import pandas as pd

def computeLeft(i):
    return np.array([i*2, i*3, i*4, i*5])

# 向量化时无需硬编码 n=4,signature 中的 'n' 会自动推断
computeLeftVectorized = np.vectorize(computeLeft, signature="()->(n)")
⚠️ 注意:np.vectorize 是语法糖,不提升底层性能;若 computeLeft 逻辑简单(如纯算术),更推荐直接用 NumPy 原生向量化(例如 np.array([indices*2, indices*3, indices*4, indices*5]).T),速度通常快数倍至数十倍。

接下来,采用 df.loc 实现精准定位赋值:

Li Python Sec Check
Li Python Sec Check

Python 安全规范检查工具:基于 CloudBase 规范、腾讯安全指南,LLM 智能分析(默认禁用,优先本地执行)

下载
results = ["val1", "val2", "val3", "val4"]
indices_to_change = [2, 5, 7, 8, 10]

# 步骤1:初始化目标列,全部填为 NaN(确保未覆盖行保持缺失)
df[results] = np.nan

# 步骤2:仅对指定索引行,批量赋值计算结果
df.loc[indices_to_change, results] = computeLeftVectorized(indices_to_change)

此方案的核心优势在于:

  • ✅ 语义清晰:loc[indices, columns] 明确表达“按标签定位行列”,避免混淆 .iloc(位置索引)与 .loc(标签索引);
  • ✅ 自动对齐:Pandas 自动将 computeLeftVectorized(indices_to_change) 返回的 (len(indices), 4) 数组,按 indices_to_change 顺序逐行映射到对应 DataFrame 行;
  • ✅ 安全容错:若 indices_to_change 中存在不在 df.index 中的标签,loc 会抛出 KeyError,便于及时发现索引不匹配问题(可通过 df.reindex() 或 df.index.isin() 预检);
  • ✅ 内存友好:无需构造全量中间数组,仅计算所需子集。

最终效果:df 中索引为 2,5,7,8,10 的行在 "val1"–"val4" 列填入 computeLeft 的计算结果(如索引 2 → [4,6,8,10]),其余所有行对应列值均为 NaN。

总结:实现选择性函数应用的黄金流程是——先初始化目标列为 NaN,再用 loc 精准定位并批量赋值。这既符合 Pandas 的向量化设计哲学,又避免了循环或掩码操作的冗余开销,是生产环境中推荐的标准实践。

热门AI工具

更多
LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

320

2025.12.04

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

232

2026.01.31

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

431

2026.04.13

FrankenPHP集成Laravel详细教程
FrankenPHP集成Laravel详细教程

本专题提供FrankenPHP集成Laravel的详细配置指南,全面解析运行原理、开发环境搭建、Caddyfile配置、Octane工作模式、数据库连接、队列任务、定时任务和生产环境优化,解决部署过程中常见的报错与兼容性问题。

0

2026.10.08

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

120

2026.09.30

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

100

2026.09.30

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

80

2026.09.30

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

80

2026.09.30

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

100

2026.09.29

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn