[SciPy-user] Array functions
Howey, David A
d.howey at imperial.ac.uk
Mon Aug 8 07:12:48 EDT 2005
I guess this is a numpy question, but how do I do truth tests on arrays?
I'm trawling around the documentation but struggling to find an answer..
I want to test if any array contains ONLY zeros.
Originally I had:
if a == 0:
# code in here
I need this to be true both for
a = 0 # ie a is just a number
and for
a = array((0,0,0,0,0))
It must return false for any array with any nonzero elements anywhere
At the moment python does this (which is wrong):
>>> vin = array((1.,0,0))
>>> vin
array([ 1., 0., 0.])
>>> if vin == 0: print "yes"
yes
>>>
Thanks
Dave
More information about the SciPy-User
mailing list