I have this iteration in a program in Matlab and want to translate it to Python, but my problem is in the parameters for 'n' and 'direction'.
for i=1:size(labels)
idx_V=[idx_V;find(y(idxUnls(trial,:))==labels(i),l/length(labels),'first')]
end
Asked
Active
Viewed 86 times
0
-
2What is the problem with these parameters? – mkrieger1 Mar 04 '21 at 21:26
-
I can't find a way to replicate them. A similar command I found was `numpy.argwhere()` but it doesn't consider those parameters. – TteDan Mar 04 '21 at 22:19
-
What do those parameters mean? What is this code supposed to do? – mkrieger1 Mar 04 '21 at 22:24
1 Answers
0
There isn't a one-to-one swap for MATLAB's find function in Python. Taking inspiration from another answer here, I would propose the following solution:
% 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]
I'm sure there is probably some trickery to think about in terms of which dimension find operates over first, compared to ndenumerate. The Python expression could also be constructed as a generator.
If you want a similar implementation, you'll have to write it yourself in Python.
aosborne
- 371
- 1
- 7
-
Also, `np.nditer` is probably the end-game if trying to implement maximum flexibility. – aosborne Mar 05 '21 at 00:59