0

0

Pandas DataFrame 行级最小值及其关联项的高效提取方法

花韻仙語

花韻仙語

发布时间:2025-10-08 11:40:02

|

991人浏览过

|

来源于php中文网

原创

Pandas DataFrame 行级最小值及其关联项的高效提取方法

本文详细介绍了如何在Pandas DataFrame中高效地查找每行的最小值,并同时提取与该最小值关联的特定列(例如,与数值列相邻的描述性列)的值。通过巧妙结合idxmin、字符串替换和NumPy风格的索引技巧,本教程提供了一种灵活且可扩展的解决方案,以满足数据分析中常见的复杂数据提取需求。

1. 引言与问题背景

在数据分析中,我们经常需要从dataframe的多个列中找出每行的最小值。然而,仅仅获取最小值本身通常不足以满足分析需求,我们可能还需要知道这个最小值是来自哪个“类别”或“描述项”。例如,如果dataframe包含成对的“值”列(value1, value2, value3)和对应的“项目”列(item1, item2, item3),我们的目标不仅是找到行级最小值,还要找出与该最小值对应的item值。

考虑以下DataFrame示例:

import pandas as pd

df = pd.DataFrame({
   'Item1': ['A', 'B', 'C', 'D'],
   'Value1': [1,4,5,7],
   'Item2': ['F', 'G', 'H', 'I'],
   'Value2': [0,4,8,12],
   'Item3': ['K', 'L', 'M', 'N'],
   'Value3': [2.7,3.4,6.2,8.1],
})

print("原始DataFrame:")
print(df)

我们的目标是生成一个包含Min_Value(每行的最小值)和Min_Item(与Min_Value对应的Item列的值)的新DataFrame,其预期结果如下:

  Item1  Value1 Item2  Value2 Item3  Value3 Min_Item  Min_Value
0     A       1     F       0     K     2.7        F        0.0
1     B       4     G       4     L     3.4        L        3.4
2     C       5     H       8     M     6.2        C        5.0
3     D       7     I      12     N     8.1        D        7.0

2. 核心思路与步骤分解

解决此问题的关键在于如何动态地从“值”列名推断出对应的“项目”列名,并利用Pandas和NumPy的强大索引能力进行数据提取。

2.1 确定参与比较的列

首先,我们需要明确哪些列参与行级最小值的比较。在这个例子中,是所有Value开头的列。

value_cols = ['Value1', 'Value2', 'Value3']

2.2 找出每行最小值的列名

使用df[value_cols].idxmin(axis=1)可以获取每行最小值所在的列名。axis=1表示按行操作。

min_value_col_names = df[value_cols].idxmin(axis=1)
print("\n每行最小值所在的列名:")
print(min_value_col_names)
# 示例输出:
# 0    Value2
# 1    Value3
# 2    Value1
# 3    Value1
# dtype: object

2.3 提取每行的最小值

有了最小值所在的列名,我们可以结合NumPy风格的索引来直接从DataFrame中提取这些值。NumPy风格的索引允许我们使用一个整数数组作为行索引,另一个整数数组作为列索引来高效地选择数据。

Civitai
Civitai

AI艺术分享平台!海量SD资源和开源模型。

下载
  1. 行索引: range(len(df)) 或 df.index 代表所有行。
  2. 列索引: df.columns.get_indexer_for(min_value_col_names) 将列名Series转换为对应的整数位置数组。
row_indices = range(len(df))
col_indices_for_min_values = df.columns.get_indexer_for(min_value_col_names)

df['Min_Value'] = df.values[row_indices, col_indices_for_min_values]
print("\n添加 Min_Value 后的 DataFrame:")
print(df)

此时,Min_Value列已经正确添加。

2.4 关键步骤:转换列名以获取关联项

现在,最关键的一步是根据min_value_col_names(例如Value2)推导出对应的Item列名(例如Item2)。这可以通过简单的字符串替换实现。

min_item_col_names = min_value_col_names.str.replace('Value', 'Item')
print("\n每行最小值对应的 Item 列名:")
print(min_item_col_names)
# 示例输出:
# 0    Item2
# 1    Item3
# 2    Item1
# 3    Item1
# dtype: object

2.5 提取每行对应的关联项

与提取最小值类似,我们再次利用NumPy风格的索引,但这次是使用转换后的min_item_col_names来获取列的整数位置。

col_indices_for_min_items = df.columns.get_indexer_for(min_item_col_names)

df['Min_Item'] = df.values[row_indices, col_indices_for_min_items]
print("\n最终结果 DataFrame:")
print(df)

3. 完整代码示例

将上述步骤整合,可以得到一个简洁高效的解决方案:

import pandas as pd

# 1. 准备数据
df = pd.DataFrame({
   'Item1': ['A', 'B', 'C', 'D'],
   'Value1': [1,4,5,7],
   'Item2': ['F', 'G', 'H', 'I'],
   'Value2': [0,4,8,12],
   'Item3': ['K', 'L', 'M', 'N'],
   'Value3': [2.7,3.4,6.2,8.1],
})

print("--- 原始 DataFrame ---")
print(df)

# 2. 定义参与比较的数值列
value_cols = ['Value1', 'Value2', 'Value3']

# 3. 获取每行最小值所在的列名 (e.g., 'Value2', 'Value3')
min_value_col_names = df[value_cols].idxmin(axis=1)

# 4. 准备行索引 (所有行)
row_indices = range(len(df))

# 5. 提取每行的最小值
# 将列名Series转换为DataFrame的实际列位置
col_indices_for_min_values = df.columns.get_indexer_for(min_value_col_names)
# 使用NumPy风格的高级索引提取值
df['Min_Value'] = df.values[row_indices, col_indices_for_min_values]

# 6. 转换列名以获取对应的关联项列名 (e.g., 'Value2' -> 'Item2')
min_item_col_names = min_value_col_names.str.replace('Value', 'Item')

# 7. 提取每行对应的关联项
# 将转换后的列名Series转换为DataFrame的实际列位置
col_indices_for_min_items = df.columns.get_indexer_for(min_item_col_names)
# 使用NumPy风格的G高级索引提取关联项
df['Min_Item'] = df.values[row_indices, col_indices_for_min_items]

print("\n--- 最终结果 DataFrame ---")
print(df)

4. 注意事项与总结

  • df.values与NumPy索引: df.values返回DataFrame的底层NumPy数组。使用df.values[row_indices, col_indices]这种NumPy风格的高级索引方式,可以高效地根据行和列的整数位置批量提取数据,这比多次使用df.loc或df.iloc要快得多。
  • df.columns.get_indexer_for(): 这个方法非常有用,它能将一个列名列表或Series转换为这些列名在DataFrame列索引中的整数位置数组,是实现NumPy高级索引的关键桥梁。
  • 列名命名规范: 本教程的解决方案依赖于ValueX和ItemX这种有规律的列名。如果你的列名没有这种规律(例如Price_A, Description_A),则需要调整字符串替换逻辑,或者维护一个从Value列名到Item列名的映射字典。
  • 性能: 这种方法利用了Pandas和NumPy的向量化操作,对于大型DataFrame来说,其性能远优于使用apply配合lambda函数或循环迭代行的方式。

通过本教程介绍的方法,您可以灵活地在Pandas DataFrame中查找行级最小值,并同时提取其对应的关联数据,从而更深入地理解和分析您的数据集。

相关专题

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

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

49

2025.12.04

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

250

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

205

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1435

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

609

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

547

2024.03.22

php中定义字符串的方式
php中定义字符串的方式

php中定义字符串的方式:单引号;双引号;heredoc语法等等。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

539

2024.04.29

go语言字符串相关教程
go语言字符串相关教程

本专题整合了go语言字符串相关教程,阅读专题下面的文章了解更多详细内容。

158

2025.07.29

php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

74

2025.12.31

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Java 教程
Java 教程

共578课时 | 40.2万人学习

国外Web开发全栈课程全集
国外Web开发全栈课程全集

共12课时 | 0.9万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号