Создание с маской в NumPy

Функция array из модуля ma принимает данные и логический массив той же формы: где в маске истина, ячейка скрыта.

Зададим цены и маску, затем выведем результат:

import numpy as np

prices = np.array([10.0, 20.0, 30.0])
hide = np.array([False, True, False])
masked_prices = np.ma.array(prices, mask=hide)
print(masked_prices) # выведет [10.0 -- 30.0]

Маску можно получить из условия через функцию masked_where: она скрывает ячейки, где условие выполняется:

import numpy as np

scores = np.array([3, 4, 5])
masked_scores = np.ma.masked_where(scores > 4, scores)
print(masked_scores) # выведет [3 4 --]

Для цен [12.0, 18.0, 24.0] скройте среднюю ячейку логической маской и выведите маскированный массив.

Для оценок [2, 4, 5, 3] скройте ячейки со значением больше 4 и выведите маскированный массив.