Функция mean
Функция mean вычисляет среднее арифметическое элементов массива.
Первым параметром функция принимает массив, для которого необходимо
рассчитать среднее значение. Вторым параметром можно указать ось,
вдоль которой производится вычисление. Третьим параметром можно
задать тип данных результата.
Синтаксис
np.mean(arr, [axis], [dtype])
Пример
Давайте вычислим среднее арифметическое всех элементов одномерного массива:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = np.mean(arr)
print(res)
Результат выполнения кода:
3.0
Пример
Вычислим среднее арифметическое элементов двумерного массива без указания оси:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.mean(arr)
print(res)
Результат выполнения кода:
3.5
Пример
Вычислим среднее арифметическое вдоль строк (по столбцам):
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.mean(arr, axis=0)
print(res)
Результат выполнения кода:
[2.5 3.5 4.5]
Пример
Вычислим среднее арифметическое вдоль столбцов (по строкам):
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.mean(arr, axis=1)
print(res)
Результат выполнения кода:
[2. 5.]
Пример
Зададим тип данных результата как целочисленный:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = np.mean(arr, dtype=np.int32)
print(res)
Результат выполнения кода:
3