From: Eric F. <ef...@ha...> - 2009-04-01 21:53:08
|
Michael Hearne wrote: > Is it possible to use multiple conditionals with the pylab.find() function? > > For example, in Matlab, I can do the following: > > x = rand(1,10); > i = find(x > 0.5 & x < 0.9); %returns the indices between 0.5 and 0.9 > > In Python: (ipython -pylab) > x = rand(1,10) > > None of the following approaches work: > #i = find(x > 0.5 and x < 0.9) > #i = find(x > 0.5 && x < 0.9) > #i = find(x > 0.5 & x < 0.9) > > I'm pretty certain there is some other way to do this using a numpy > method, and if that's all there is, then I will happily use it. > > However, if the goal of the pylab module is to achieve (some level of) > compatibility with Matlab, then may I suggest this sort of functionality > be added to the pylab.find() function? This is a fundamental python/numpy problem. The workaround is x = rand(10) i = find((x > 0.5) & (x < 0.9)) Note the explicit parentheses to make the conditionals evaluated before the bitwise and. The real problem is that we are stuck using bitwise and as a surrogate for logical and. This problem is not going away any time soon, although proposed solutions have been discussed. Also note that in numpy we are using a 1-D array. If you really want the 2-D array, then you will have to index it as x[:,i]. Longer term, the recommendation is that you move away from trying to write matlab code using pylab, and instead use numpy idioms directly. import numpy as np x = np.random.rand(10) y = x[(x > 0.5) & (x < 0.9)] or if you really want the indices: i = np.nonzero((x > 0.5) & (x < 0.9))[0] Eric > > Thanks, > > Mike > > > ------------------------------------------------------------------------------ > _______________________________________________ > Matplotlib-users mailing list > Mat...@li... > https://lists.sourceforge.net/lists/listinfo/matplotlib-users |