How do I calculate the mean of a NumPy array in Python?

How do I calculate the mean of a NumPy array in Python?

You can use the numpy.mean() function to calculate the mean of a NumPy array. Just pass your array as an argument.

Can you show me an example?

Sure! Here's a simple example:

import numpy as np

array = np.array([1, 2, 3, 4, 5])
mean_value = np.mean(array)
print(mean_value)  # Output: 3.0

What if I have a 2D array?

For a 2D array, you can specify the axis along which to calculate the mean. For example:

array_2d = np.array([[1, 2, 3], [4, 5, 6]])
mean_value_axis0 = np.mean(array_2d, axis=0)  # Mean along columns
mean_value_axis1 = np.mean(array_2d, axis=1)  # Mean along rows