NumPy: вычисления и устройство массивов

Объединение массивов: 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)
  • Тест 1
    Ввод
    3 4 2
    Ожидаемый вывод
    (2, 3, 4)
    (3, 2, 4)
    (3, 4, 2)
Решение
Как проверяется решение

Сравнение вывода: Правила исходной проверяющей системы. Интерактивный запуск не влияет на оценку. Лимит сессии — 5 минут, процессорного времени — 10 секунд.

Отправьте решение, чтобы увидеть результаты тестов.