Функция rint
Функция rint округляет каждый элемент массива до ближайшего целого числа.
Она применяется поэлементно к массиву чисел и возвращает новый массив с округлёнными значениями.
Округление выполняется по правилам банковского округления: если дробная часть равна ровно 0.5,
то число округляется до ближайшего чётного целого.
Функция принимает массив в качестве единственного обязательного аргумента
и возвращает массив того же размера с типом float64.
Синтаксис
np.rint(arr, out=None, where=True)
Пример
Давайте округлим одномерный массив с положительными числами:
import numpy as np
arr = np.array([1.2, 2.7, 3.5, 4.8, 5.1])
res = np.rint(arr)
print(res)
Результат выполнения кода:
[1. 3. 4. 5. 5.]
Пример
Рассмотрим работу функции с отрицательными числами:
import numpy as np
arr = np.array([-1.2, -2.7, -3.5, -4.8, -5.1])
res = np.rint(arr)
print(res)
Результат выполнения кода:
[-1. -3. -4. -5. -5.]
Пример
Применим функцию к двумерному массиву:
import numpy as np
arr = np.array([[0.1, 0.6, 1.5], [2.3, 3.7, 4.5]])
res = np.rint(arr)
print(res)
Результат выполнения кода:
[[0. 1. 2.]
[2. 4. 4.]]
Пример
Используем параметр out для сохранения результата в уже существующий массив:
import numpy as np
arr = np.array([1.2, 2.7, 3.5, 4.8, 5.1])
out_arr = np.empty_like(arr)
np.rint(arr, out=out_arr)
print(out_arr)
Результат выполнения кода:
[1. 3. 4. 5. 5.]
Пример
Применим функцию с условием через параметр where для выборочного округления:
import numpy as np
arr = np.array([1.2, 2.7, 3.5, 4.8, 5.1])
mask = arr > 2.0
res = np.rint(arr, where=mask)
print(res)
Результат выполнения кода:
[0. 3. 4. 5. 5.]
