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

Question

在 MATALAB 中我可以轻松编写：

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

最多返回与 X 的非零条目对应的最后 k 个索引。

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

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

Answer 1

等价表达式是

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

使用切片可以替换 'last' 参数的使用。

Answer 2

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