理解numpy.where

• numpy.where(condition[, x, y])

  Return elements chosen from x or y depending on condition.

  >>> a = np.arange(10)
  >>> np.where(a<5, a, 10*a)
  array([0,1,2,3,4,50,60,70,80,90])