Задача курса
NumPy stack: axis=0, 1 и 2 и форма результата
Курс «NumPy и pandas: практический тренажёр» · урок «Объединение массивов: concatenate и stack»
Условие
np.stack по трём позициям оси: предсказание и проверка shape
Что уже дано
Проверяющая система запускает вашу программу и подаёт данные через stdin. Никаких заранее созданных переменных нет — вы читаете всё сами.
Что нужно сделать
Напишите программу, которая:
- Читает из
stdinтри целых числа в одной строке:R,C,n— число строк, столбцов и количество одинаковых двумерных массивов. - Создаёт
nдвумерных массивов одинаковой формы(R, C), гдеi-й массив равенnp.arange(i * R * C, (i + 1) * R * C).reshape(R, C)дляiот0доn-1. - Складывает все
nмассивов черезnp.stackсaxis=0, затем сaxis=1, затем сaxis=2. Сохраняет результаты в переменныеs0,s1,s2. - Выводит в
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
Без регистрации · результат не сохраняется
