R语言which()与which.min()的Python等效实现方法咨询

我懂你想要在NumPy里实现和R中类似的索引查找功能，而且你已经发现Python列表的index()方法在NumPy数组上不适用——这很正常，NumPy有自己专门的工具来处理这类数组索引问题。下面我一步步给你演示怎么实现：

1. 获取满足条件的所有元素索引（对应R的which(x==2)） #

首先，我们用NumPy的np.where()函数来定位所有符合条件的元素位置。以下是完整示例：

import numpy as np

# 定义你的数组
x = np.array([1,2,3,4,0,1,2,3,4,11])

# 获取所有x==2的索引
matching_indices = np.where(x == 2)[0]
print(matching_indices)  # 输出: [1 6]

这里要注意：np.where()返回的是一个元组（因为要兼容多维数组），所以我们取索引[0]来得到一维数组的索引列表。如果只需要第一个满足条件的索引（对应R的min(which(x==2))），直接取数组的第一个元素就行：

first_match = matching_indices[0]
print(first_match)  # 输出: 1

2. 获取最小值的索引（对应R的which.min()） #

NumPy里直接用np.argmin()函数就能实现这个功能，它会返回数组中最小值第一次出现的索引：

min_value_index = np.argmin(x)
print(min_value_index)  # 输出: 4（因为x[4]是0，是数组中的最小值）

为什么你之前的尝试报错？ #

你用x[x>2].index()时出现'numpy.ndarray' object has no attribute 'index'的错误，是因为x[x>2]返回的是一个NumPy数组，而index()是Python原生列表的方法，NumPy数组并没有这个属性。所以处理NumPy数组的索引问题，一定要用NumPy提供的专用函数哦。

内容的提问来源于stack exchange，提问作者Hadij