Pythonist

Кодинг-марафон. Задача 9.

Напишите функцию, которая будет возвращать строку, представляющую собой ASCII-рисунок ромба. Функция должна иметь сигнатуру def diamond(letter: str, background: str=' ') -> str:

Функция получает на вход одну букву латинского алфавита в верхнем регистре и один символ, который будет использован в качестве фона для рисунка:

    >> diamond('A')
    A

    >> diamond('B')
     A
    B B
     A

    >> diamond('D', '-')
    ---A---
    --B-B--
    -C---C-
    D-----D
    -C---C-
    --B-B--
    ---A---

КАЖДАЯ строка рисунка заканчивается символом \n, на вход может подаваться любая из 26 букв латинского алфавита и любой ASCII-символ как параметр background. Предполагается, что ввод всегда валиден, не нужно проверок и исключений.

Решение

from string import ascii_uppercase


def diamond(letter: str, background: str = ' ') -> str:
    length = ascii_uppercase.index(letter)
    template = [[background for __ in range(length * 2 + 1)] for _ in range(length + 1)]
    template[0][length] = 'A'
    for index in range(1, len(template)):
        template[index][length - index] = ascii_uppercase[index]
        template[index][-(length - index) - 1] = ascii_uppercase[index]
    template.extend(list(reversed(template))[1:])
    return '\n'.join(''.join(_) for _ in template) + '\n'