怎样在Python中使用Pandas根据行索引快速切片和抽样

风萱酱_4872

风萱酱_4872

2026-10-24

647人浏览

原创

应优先使用.loc按索引值精确切片,而非.iloc(后者按位置编号);.loc支持字符串/日期索引、闭区间切片、布尔索引;抽样首选sample()以保留索引并提升性能;索引未排序时.loc切片需先sort_index()或改用布尔索引。

怎样在python中使用pandas根据行索引快速切片和抽样

用 .loc 按行索引精确切片,别用 .iloc

如果索引是字符串、日期或非连续整数,.iloc 会按位置编号取数,和你想的“按索引值取”完全不是一回事。比如 df.iloc[0] 总是取第一行,哪怕索引是 '2023-01-01';而 df.loc['2023-01-01'] 才真按索引值匹配。

常见错误:把 .loc 当成“带标签的 .iloc”,结果遇到重复索引时报 KeyError 或返回多行却没意识到。

  • 单个索引值:df.loc['A'](返回 Series)或 df.loc[['A']](返回 DataFrame)
  • 切片(闭区间):df.loc['A':'C'] —— 注意这是按索引顺序切,不是按原始行序
  • 布尔索引配合:df.loc[df.index.str.startswith('2023')],比先重置索引再 .iloc 快得多

抽样时优先用 sample() 而非 np.random.choice() + .iloc

df.sample(n=100, random_state=42) 内部直接操作索引,不触发数据拷贝,速度比手动构造位置列表再 .iloc 高 3–5 倍(尤其对百万行以上 DataFrame)。它默认按行抽样,且天然支持权重(weights=df['score'])和分层(stratify=df['category'])。

容易踩的坑:用 df.iloc[np.random.choice(len(df), size=100)] 会丢失原始索引,且无法利用 Pandas 的索引加速路径。

python学习
python学习

Python 编程学习助手 skill,面向零基础用户。 当用户表达以下意图时触发: - 学习 Python 基础概念(变量、数据类型、循环、函数、类等) - 请求 Python 练习题或测验 - 请求审查或调试 Python 代码 - 请求 Python 项目实战指导(爬虫、数据分析、Web 开发等入门项目) - 泛化表述如"教我 Python""我想学编程""Python 怎么写""帮我学 Python" 此 skill

下载
  • 按索引抽样(保留原索引):df.sample(n=100, axis=0)
  • 等价但更慢的写法:df.iloc[np.random.permutation(len(df))[:100]]
  • 想确保抽到特定索引子集?先用 .loc 过滤再 .sample,别在全量数据上随机后筛选

索引未排序时,.loc 切片可能意外返回空或报错

df.loc['2023-01-01':'2023-01-05'] 要求索引已升序排列,否则 Pandas 会跳过中间未排序的段,甚至抛 KeyError(如索引是 ['2023-01-05', '2023-01-01'])。这不是 bug,是设计行为——Pandas 把切片当作有序查找优化路径。

  • 检查是否有序:df.index.is_monotonic_increasing
  • 安全做法:df.sort_index().loc['2023-01-01':'2023-01-05'],但注意 sort_index() 有开销
  • 替代方案(不排序):df[df.index >= '2023-01-01'][df.index ,逻辑清晰但两次布尔索引略慢

大文件场景下,避免反复 .loc 查找,先缓存索引位置

如果你要对同一组索引做多次切片(比如循环中查 df.loc[key]),每次调用 .loc 都会重建哈希映射或二分搜索结构。对千万级索引,这能差出 10 倍时间。

  • 提前构建索引映射:idx_map = {k: i for i, k in enumerate(df.index)},然后用 df.iloc[idx_map['A']]
  • 更推荐:df.index.get_loc('A') 返回位置,比字典查更快且内存省
  • 批量查找多个键:df.index.get_indexer(['A', 'B', 'C']),返回位置数组,一次搞定

索引类型影响性能:纯整数索引走快速路径,字符串或 datetime 索引依赖哈希或树结构,重复索引会退化为线性扫描。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1791

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4724

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1749

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

27997

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3367

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3387

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1203

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2623

5

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习