如何将 MATLAB 的命令 find 转换为 Python？ [复制]

Question

我在 Matlab 的一个程序中有这个迭代，想把它翻译成 Python，但我的问题在于 'n' 和 'direction' 的参数。

for i=1:size(labels)
    idx_V=[idx_V;find(y(idxUnls(trial,:))==labels(i),l/length(labels),'first')]
end

Answer 1

在 Python 中，MATLAB 的 find 函数没有一对一的交换。受到another answer here的启发，我提出以下解决方案：

% MATLAB
inds = find(myarray == condition, n, 'first')

# Python
import numpy as np
inds = [ind for (ind, val) in np.ndenumerate(myarray == condition) if val]
inds = inds[0:n]

我敢肯定，与ndenumerate 相比，考虑到find 哪个维度首先运行，我确信可能需要考虑一些技巧。 Python 表达式也可以构造为生成器。

如果您想要类似的实现，则必须自己用 Python 编写。