diff options
Diffstat (limited to 'numpy/core/fromnumeric.py')
| -rw-r--r-- | numpy/core/fromnumeric.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/numpy/core/fromnumeric.py b/numpy/core/fromnumeric.py index ed1eeb908..f4c7af88a 100644 --- a/numpy/core/fromnumeric.py +++ b/numpy/core/fromnumeric.py @@ -3036,14 +3036,17 @@ def prod(a, axis=None, dtype=None, out=None, keepdims=np._NoValue, Even when the input array is two-dimensional: - >>> np.prod([[1.,2.],[3.,4.]]) + >>> a = np.array([[1., 2.], [3., 4.]]) + >>> np.prod(a) 24.0 But we can also specify the axis over which to multiply: - >>> np.prod([[1.,2.],[3.,4.]], axis=1) + >>> np.prod(a, axis=1) array([ 2., 12.]) - + >>> np.prod(a, axis=0) + array([3., 8.]) + Or select specific elements to include: >>> np.prod([1., np.nan, 3.], where=[True, False, True]) |