Don't like ads? PRO users don't see any ads ;-)
Public Pastes
Guest

Untitled

By: a guest on May 5th, 2012  |  syntax: None  |  size: 0.46 KB  |  hits: 11  |  expires: Never
download  |  raw  |  embed  |  report abuse  |  print
Text below is selected. Please press Ctrl+C to copy to your clipboard. (⌘+C on Mac)
  1. Where function in python returns nothing
  2. a = array([1,5,7])
  3.        
  4. where(a==8)
  5.        
  6. (array([], dtype=int64),)
  7.        
  8. def where0(vec):
  9.     a = where(vec)
  10.     return a if a[0] else 0
  11.     # The return above is equivalent to:
  12.     # if len(a[0]) == 0:
  13.     #     return 0  # or whatever you like
  14.     # else:
  15.     #     return a
  16.  
  17. a = array([1,5,7])
  18. print where0(a==8)
  19.        
  20. a = array([1,5,7])
  21.  result = where(a==8)
  22.  
  23.  if result[0] != 0:
  24.      doFancyStuff(result)
  25.  else:
  26.      print "bump"
create a new version of this paste RAW Paste Data