如何高效查找 NumPy 数组中多个值的行索引？

在 NumPy 数组中查找多个值的行索引

给定一个数组 X 和一组目标行 searched_values，任务是检索相应的行索引。使用各种 NumPy 函数可以有效地解决这个问题。

方法 1：广播

使用广播的简单方法：

np.where((X==searched_values[:,None]).all(-1))[1]

方法 2：内存高效转换

为了内存效率，将每一行转换为唯一的线性索引并使用np.in1d:

dims = X.max(0)+1
out = np.where(np.in1d(np.ravel_multi_index(X.T,dims),\
                    np.ravel_multi_index(searched_values.T,dims)))[0]

方法 3：内存高效搜索

使用 np.searchsorted 的另一种内存高效解决方案：

dims = X.max(0)+1
X1D = np.ravel_multi_index(X.T,dims)
searched_valuesID = np.ravel_multi_index(searched_values.T,dims)
sidx = X1D.argsort()
out = sidx[np.searchsorted(X1D,searched_valuesID,sorter=sidx)]

注意： 这种方法假设 searched_values 中的每一行都与X.

以上是如何高效查找 NumPy 数组中多个值的行索引？的详细内容。更多信息请关注PHP中文网其他相关文章！