本教程详细介绍了如何在 Pandas DataFrame 中高效地查找每一行的最小值，并进一步提取与该最小值对应的非数值型关联列值（例如，项目名称）。文章通过一个具体的示例，展示了如何利用 idxmin、str.replace 和 get_indexer_for 等 Pandas 功能，以简洁且性能优越的方式实现这一常见的数据处理需求，避免了复杂的迭代或 apply 操作。
引言
在数据分析中，我们经常需要从 dataframe 的多列中找出每行的最小值。然而，除了最小值本身，有时我们还需要知道是哪一列或哪个“项目”对应了这个最小值。例如，在一个包含多个产品值及其对应产品名称的 dataframe 中，我们可能需要找出每行中最低的价格以及对应的产品名称。本教程将展示如何利用 pandas 的强大功能，以一种高效且优雅的方式解决这个问题。
示例数据准备
首先，我们创建一个示例 Pandas DataFrame，其中包含数值列（如 Value1, Value2, Value3）和与之关联的非数值型列（如 Item1, Item2, Item3）。
import pandas as pddf = 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)
输出的原始 DataFrame 如下：
原始 DataFrame: Item1 Value1 Item2 Value2 Item3 Value30 A 1 F 0 K 2.71 B 4 G 4 L 3.42 C 5 H 8 M 6.23 D 7 I 12 N 8.1
我们的目标是添加两列：Min_Value (每行的最小值) 和 Min_Item (对应最小值的项目名称)。
步骤一：查找每行的最小值及其所在列名
首先，我们需要确定哪些列参与最小值的比较。在这个例子中，是 Value1, Value2, Value3。
# 定义参与比较的数值列value_cols = ['Value1', 'Value2', 'Value3']# 使用 idxmin(axis=1) 找出每行最小值所在的列名# 结果是一个 Series，其索引是 DataFrame 的行索引，值是最小值所在列的名称（如 'Value1', 'Value2'）min_value_col_names = df[value_cols].idxmin(axis=1)# 获取行索引序列，用于后续的基于 NumPy 数组的索引row_indices = range(len(df))# 利用 df.values 和 get_indexer_for 提取最小值# df.values 将 DataFrame 转换为 NumPy 数组，可以进行高效的整数位置索引# df.columns.get_indexer_for(min_value_col_names) 将列名 Series 转换为对应的整数列索引 Seriesdf['Min_Value'] = df.values[row_indices, df.columns.get_indexer_for(min_value_col_names)]print("n添加 Min_Value 后的 DataFrame:")print(df)
此时 DataFrame 会新增 Min_Value 列：
添加 Min_Value 后的 DataFrame: Item1 Value1 Item2 Value2 Item3 Value3 Min_Value0 A 1 F 0 K 2.7 0.01 B 4 G 4 L 3.4 3.42 C 5 H 8 M 6.2 5.03 D 7 I 12 N 8.1 7.0
步骤二：提取对应最小值的关联列值（项目名称）
关键在于如何将 min_value_col_names (例如 ‘Value1’) 转换为其对应的项目列名 (例如 ‘Item1’)。这里我们可以利用字符串替换功能。
# 将最小值所在列的名称（如 'Value1'）转换为对应的项目列名（如 'Item1'）# 假设项目列名和值列名之间存在 'Value' 到 'Item' 的简单映射关系min_item_col_names = min_value_col_names.str.replace('Value', 'Item')# 再次利用 df.values 和 get_indexer_for 提取对应的项目名称df['Min_Item'] = df.values[row_indices, df.columns.get_indexer_for(min_item_col_names)]print("n最终结果 DataFrame:")print(df)
最终输出的 DataFrame 将包含 Min_Value 和 Min_Item 两列，符合我们的预期：
最终结果 DataFrame: Item1 Value1 Item2 Value2 Item3 Value3 Min_Value Min_Item0 A 1 F 0 K 2.7 0.0 F1 B 4 G 4 L 3.4 3.4 L2 C 5 H 8 M 6.2 5.0 C3 D 7 I 12 N 8.1 7.0 D
完整代码示例
为了方便理解和使用，下面是实现上述功能的完整代码块：
import pandas as pd# 示例 DataFramedf = 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],})# 1. 定义参与比较的数值列value_cols = ['Value1', 'Value2', 'Value3']# 2. 获取每行最小值所在列的名称# axis=1 表示按行操作min_value_col_names = df[value_cols].idxmin(axis=1)# 3. 获取行索引序列，用于基于 NumPy 数组的索引row_indices = range(len(df))# 4. 提取每行的最小值# df.values 将 DataFrame 转换为 NumPy 数组，支持高效的整数位置索引# df.columns.get_indexer_for() 将列名 Series 转换为对应的整数列索引 Seriesdf['Min_Value'] = df.values[row_indices, df.columns.get_indexer_for(min_value_col_names)]# 5. 将最小值所在列的名称转换为对应的项目列名# 假设项目列名和值列名之间存在 'Value' 到 'Item' 的简单映射关系min_item_col_names = min_value_col_names.str.replace('Value', 'Item')# 6. 提取对应的项目名称df['Min_Item'] = df.values[row_indices, df.columns.get_indexer_for(min_item_col_names)]print("最终处理结果 DataFrame:")print(df)
关键概念解析
df[cols].idxmin(axis=1): 这个方法用于找出指定列 cols 中每行最小值的列名。axis=1 指示按行操作。它返回一个 Series，其索引是原始 DataFrame 的行索引，值是最小值所在列的名称。df.values: 将 Pandas DataFrame 转换为底层的 NumPy 数组。NumPy 数组支持非常高效的整数位置索引，这对于大规模数据处理至关重要。df.columns.get_indexer_for(labels): 这个方法接收一个标签（列名或索引名）列表或 Series，并返回这些标签在 DataFrame 列索引中的整数位置。例如，df.columns.get_indexer_for([‘Value1’, ‘Item2’]) 可能会返回 [1, 2]（如果 ‘Value1′ 是第1列，’Item2’ 是第2列）。这使得我们可以使用整数位置来高效地从 df.values 中提取数据。Series.str.replace(‘old’, ‘new’): 这是 Pandas Series 字符串方法的强大功能之一，允许我们对 Series 中的每个字符串元素执行字符串替换操作。在此例中，它将 ‘ValueX’ 转换为 ‘ItemX’，实现了列名映射。
注意事项与总结
命名约定: 本教程的方法高度依赖于数值列 (ValueX) 和关联列 (ItemX) 之间的命名约定。如果你的列名没有这种规律性（例如，PriceA, ProductA_Name），你需要采用更复杂的映射逻辑（如使用字典进行 map 操作）。性能: 这种方法通过利用 Pandas 和 NumPy 的底层优化，避免了显式的 Python 循环或 df.apply() 函数（尤其是在处理大型 DataFrame 时，apply 可能会较慢），因此在性能上非常高效。灵活性: 这种模式可以推广到其他类似的需求，例如查找最大值及其关联列，或者根据特定条件查找值及其关联信息。
通过上述步骤，我们不仅找到了每行的最小值，还成功地提取了与之对应的关联信息，这在实际数据分析工作中是非常有用的技巧。掌握这种高效的 Pandas 索引和字符串处理方法，将大大提升你的数据处理能力。
以上就是Pandas DataFrame：高效获取行级最小值及其对应关联列值的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1376707.html
微信扫一扫 
支付宝扫一扫 
