MATLAB find 函数转换为 python - 3 个输入参数版本

MATLAB find function translate to python - 3 input arguments version

在MATALAB中我可以轻松写出:

ind = find(X, k, 'last')

最多 returns 对应于 X 的非零条目的最后 k 个索引。

Numpy 具有 numpy.nonzero 函数,仅适用于 MATALAB find(X),单参数版本。

用 3 个参数翻译 find 的 numpythonic 方法是什么?

您将需要通过结果的数组切片来应用最后一个或第一个 k。

等效表达式是

# importing numpy as np and
# assign a ndarray to x
ind = np.nonzero(x)[-k:]

使用切片代替 'last' 参数。