Folks, the doc for `where` says "x and y need to have the same shape as condition" http://docs.scipy.org/doc/numpy-dev/reference/generated/numpy.where.html But surely "where is equivalent to: [xv if c else yv for (c,xv,yv) in zip(condition,x,y)]" holds as long as len(condition) == len(x) == len(y) ? And `condition` can be broadcast ? n = 3 all01 = np.array([ t for t in np.ndindex( n * (2,) )]) # 000 001 ... x = np.zeros(n) y = np.ones(n) w = np.where( all01, y, x ) # 2^n x n
Can anyone please help me understand `where` / extend "where is equivalent to ..." ? Thanks, cheers -- denis _______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@scipy.org http://mail.scipy.org/mailman/listinfo/numpy-discussion