Задачи NumPy онлайн: массивы и вычисления с проверкой

Здесь можно практиковать NumPy на задачах с проверкой кода: от создания ndarray до операций над матрицами. В подборке есть и тесты, где нужно предсказать поведение массива.

С чего начать

Сначала научитесь читать shape, ndim, size и dtype. Затем выберите строку, столбец и прямоугольный блок матрицы. После этого переходите к маскам, вычислениям по осям и broadcasting. Такой порядок помогает понимать ошибки формы, а не подбирать выражения наугад.

Небольшой пример

import numpy as np
a = np.array([[1, 3, 5], [2, 4, 6]])
print(a.shape)
print(a.sum(axis=0).tolist())
print(a.sum(axis=1).tolist())

Результат:

(2, 3)
[3, 7, 11]
[9, 12]

axis=0 убирает ось строк и оставляет сумму для каждого столбца. axis=1 убирает ось столбцов и оставляет сумму для каждой строки. Сопоставьте длину результата с формой массива.

Выберите тему

Как решать

Откройте упражнение, прочитайте формат входа и вывода и запустите проверку своего кода. Условие конкретной задачи важнее общего примера выше. Для полного маршрута с объяснениями есть бесплатный тренажёр NumPy и pandas.

Задачи NumPy с проверкой: массивы, срезы, axis и broadcasting