Функция NumPy mean() в Python

Чтобы вычислить среднее значение элементов в массиве в целом, вдоль оси или нескольких осей, используйте функцию numpy.mean() в Python.

В этом руководстве мы рассмотрим следующие примеры, используя функцию:

• Среднее значение всех элементов в массиве.
• Среднее значение элементов массива по оси.
• Среднее значение элементов массива по нескольким осям.

Пример 1: среднее значение всех элементов в массиве

В этом примере мы берем 2D-массив NumPy и вычисляем среднее значение массива.

import numpy as np

#initialize array
A = np.array([[2, 1], [5, 4]])

#compute mean
output = np.mean(A)

print(output)

Выход

3.0

Иметь в виду

Mean = (2 + 1 + 5 + 4)/4
     = 12/4
     = 3.0

Пример 2: Среднее значение элементов массива NumPy по оси

В этом примере мы берем 2D-массив NumPy и вычисляем среднее значение элементов вдоль одной, скажем, оси = 0.

Передайте указанную ось аргументов функции mean(), как показано ниже.

import numpy as np

#initialize array
A = np.array([[2, 1], [5, 4]])

#compute mean
output = np.mean(A, axis=0)

print(output)

Выход

[3.5 2.5]

Понимание оси

Поскольку мы предоставили axis = 0 в качестве аргумента, эта ось сокращается для вычисления среднего значения вдоль этой оси, сохраняя другую ось.

       [    [2, 1],  [5, 4]   ]
axis:  0    1        1 

[2, 1] and [5, 4] are the elements of axis=0.

Иметь в виду

Mean = ([2, 1] + [5, 4])/2
     = [(2 + 5)/2, (1 + 4)/2]
     = [7/2, 5/2]
     = [3.5, 2.5]

Пример 3: Среднее значение элементов массива NumPy по нескольким осям

В этом примере мы берем трехмерный массив NumPy, чтобы мы могли указать как минимум две оси, и вычисляем среднее значение массива.

Передайте указанную ось аргументов с кортежем осей в функцию mean(), как показано ниже.

import numpy as np

#initialize array
A = np.array([[[2, 1], [5, 4]], [[3, 9], [6, 8]]])

#compute mean
output = np.mean(A, axis=(0, 1))

print(output)

Выход

[4.  5.5]

Понимание оси

Поскольку мы предоставили axis = (01 1) в качестве аргумента, эти оси сокращаются для вычисления среднего значения вдоль этой оси, сохраняя другую ось. которая является осью: 2.

       [    [  [2, 1], [5, 4]], [  [3, 9], [6, 8] ]  ]
axis:  0    1  2       2        1  2       2

[[2, 1], [5, 4]] and [[3, 9], [6, 8]] are the elements of axis=0.
[2, 1], [5, 4], [3, 9], [6, 8] are the elements of axis=1.

Иметь в виду

Mean = ([2, 1] + [5, 4] + [3, 9] + [6, 8])/4
     = [(2 + 5 + 3 + 6)/4, (1 + 4 + 9 + 8)/4]
     = [4.0, 5.5]

Резюме

В этом руководстве примеров Python мы узнали, как найти среднее значение Numpy, всего массива, вдоль оси или вдоль нескольких осей с помощью хорошо подробных примеров программ Python.