Задача курса

NumPy stack: axis=0, 1 и 2 и форма результата

Курс «NumPy и pandas: практический тренажёр» · урок «Объединение массивов: concatenate и stack»

Условие

np.stack по трём позициям оси: предсказание и проверка shape

Что уже дано

Проверяющая система запускает вашу программу и подаёт данные через stdin. Никаких заранее созданных переменных нет — вы читаете всё сами.

Что нужно сделать

Напишите программу, которая:

  1. Читает из stdin три целых числа в одной строке: R, C, n — число строк, столбцов и количество одинаковых двумерных массивов.
  2. Создаёт n двумерных массивов одинаковой формы (R, C), где i-й массив равен np.arange(i * R * C, (i + 1) * R * C).reshape(R, C) для i от 0 до n-1.
  3. Складывает все n массивов через np.stack с axis=0, затем с axis=1, затем с axis=2. Сохраняет результаты в переменные s0, s1, s2.
  4. Выводит в stdout три строки — shape каждого результата в виде Python-кортежа:
  • строка 1: shape s0;
  • строка 2: shape s1;
  • строка 3: shape s2.

Ввод и вывод

Программа читает stdin и пишет в stdout.

Формат ввода

Одна строка: три целых числа R C n, разделённых пробелами. Гарантируется R >= 2, C >= 2, n >= 2.

Формат вывода

Три строки. Каждая — shape соответствующего результата np.stack в виде Python-кортежа, например (3, 2, 4).

Примеры

Пример 1

Ввод
3 4 2
Ожидаемый результат
(2, 3, 4)
(3, 2, 4)
(3, 4, 2)

Попробуйте решить

РешениеPython
Без регистрации · результат не сохраняется