[Numpy-discussion] sum and prod
warren.weckesser at enthought.com
Sat Sep 8 18:10:16 EDT 2012
On Sat, Sep 8, 2012 at 4:56 PM, nicky van foreest <vanforeest at gmail.com>wrote:
> I ran the following code:
> args = np.array([4,8])
> print np.sum( (arg > 0) for arg in args)
> print np.sum([(arg > 0) for arg in args])
> print np.prod( (arg > 0) for arg in args)
> print np.prod([(arg > 0) for arg in args])
> with this result:
I get 2 here, not 1 (numpy version 1.6.1).
> <generator object <genexpr> at 0x1c70410>
> Is the difference between prod and sum intentional? I would expect
> that numpy.prod would also work on a generator, just like numpy.sum.
Whatever the correct result may be, I would expect them to have the same
behavior with respect to a generator argument.
> BTW: the last line does what I need: the product over the truth values
> of all elements of args. Is there perhaps a nicer (conciser) way to
> achieve this? Thanks.
In : np.all(args > 0)
> NumPy-Discussion mailing list
> NumPy-Discussion at scipy.org
-------------- next part --------------
An HTML attachment was scrubbed...
More information about the NumPy-Discussion