NumPy stack: axis=0, 1 и 2 и форма результата
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).
Пример
- Ввод
3 4 2
- Вывод
(2, 3, 4) (3, 2, 4) (3, 4, 2)
РешениеPython
Без регистрации. Гостевая проверка не сохраняет прогресс. Для сохранения перейдите к уроку.
Отправьте решение, чтобы увидеть результаты тестов.
Связанные задачи
- NumPy concatenate и stack: чем отличаются shape и axis
- dtype NumPy: целые и вещественные массивы, типы элементов
- NumPy shape, ndim, size и dtype: атрибуты массива и задача
- Срезы двумерного массива NumPy: строки, столбцы и шаг
- NumPy reshape и transpose: разница на примере и задача
- NumPy newaxis и squeeze: вектор-строка и вектор-столбец
