Calculate mean across dimension in a 2D array

by

a.mean() takes an axis argument:

In [1]: import numpy as np

In [2]: a = np.array([[40, 10], [50, 11]])

In [3]: a.mean(axis=1)     # to take the mean of each row
Out[3]: array([ 25. ,  30.5])

In [4]: a.mean(axis=0)     # to take the mean of each col
Out[4]: array([ 45. ,  10.5])

Or, as a standalone function:

In [5]: np.mean(a, axis=1)
Out[5]: array([ 25. ,  30.5])

The reason your slicing wasn’t working is because this is the syntax for slicing:

In [6]: a[:,0].mean() # first column
Out[6]: 45.0

In [7]: a[:,1].mean() # second column
Out[7]: 10.5

More Related Contents:

  1. Using numpy to build an array of all combinations of two arrays
  2. Understanding NumPy’s einsum
  3. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  4. Why is numpy’s einsum faster than numpy’s built in functions?
  5. Selecting specific rows and columns from NumPy array
  6. Convert a 1D array to a 2D array in numpy
  7. Subsetting a 2D numpy array
  8. Check if two 3D numpy arrays contain overlapping 2D arrays
  9. Selecting Random Windows from Multidimensional Numpy Array Rows
  10. How to split a numpy array in fixed size chunks with and without overlap?
  11. How to find the index of a value in 2d array in Python?
  12. What is the difference between ndarray and array in NumPy?
  13. Python: how to store a numpy multidimensional array in PyTables?
  14. Access n-th dimension in python [duplicate]
  15. How does the axis parameter from NumPy work?
  16. Mapping a NumPy array in place
  17. How to convert a 3d numpy array to 2d
  18. numpy matrix vector multiplication [duplicate]
  19. How to apply a disc shaped mask to a NumPy array?
  20. How to remove specific elements in a numpy array
  21. Find nearest indices for one array against all values in another array – Python / NumPy
  22. Using Numpy Vectorize on Functions that Return Vectors
  23. How to extend an array in-place in Numpy?
  24. “In” operator for numpy arrays?
  25. np arrays being immutable – “assignment destination is read-only”
  26. Numpy individual element access slower than for lists
  27. How to turn a video into numpy array?
  28. Why using an array as an index changes the shape of a multidimensional ndarray?
  29. Why is B = numpy.dot(A,x) so much slower looping through doing B[i,:,:] = numpy.dot(A[i,:,:],x) )?
  30. Efficient dot products of large memory-mapped arrays

Leave a Comment