Python3 как возвести в степень каждую цифру натурального числа?

Question

[Евгений Сноу]

Решаю задачки на codewars, попалась такая которая сломала меня)
Суть в чем: нужна функция которая возводит в квадрат цифры натурального числа, каждую по очереди.
например 91 будет 811, поэтому 9 в квадрате будет 81, а 1 в квадрате будет 1.
Я уже второй день пытаюсь и тяму не хватает.
Кода нет еще, необходимо чтобы на входе было целое число и на выходе тоже в условиях задачи.
Как бы я не пытался использовать кортежи, списки, я утыкаюсь в то что число не может быть итерируемым объектом и никакого for in не получится. Помогите советом или примером. За ранее благодарю
def square_digits(num):
???

Comments

[AWEme]

Не можешь итерироваться по числу - итерируйся по строке, например. Или можешь разбить число на отдельные цифры, путём деления по модулю числа на основание системы счисления, и сложить всё это в список, а потом уже итерироваться.

[Евгений Сноу]

AWEme, спасибо за совет а можешь пояснить "путём деления по модулю числа на основание системы счисления", на каком то примере?

[AWEme]

Евгений Сноу,

>>> number = 91
>>> digits = []
>>> while number:
... 	number, mod = divmod(number, 10)
... 	digits.append(mod)
... 
>>> digits
[1, 9]

Answer 1

[Владимир Куц]

На скорую руку - простейшее решение, если я правильно понял суть задачи:

def square_digits(number):
    out = ''
    for im in str(number):
        out += str(int(im)**2)
    return int(out)

>>> square_digits(91)
811

Comments

[Евгений Сноу]

out=''
то есть правильно ли я понял, ты создаешь пустую строку
затем циклом идешь по строке number
затем out+=str(int(im)**2) - а так разве можно?:)
Можешь прокоментировать пожалуйста, то есть строка out прибавляет строку в которой числа...
Мой мозг не может переварить) я еще начинающий програмист:)

Но решение удалось. но я не полностью осознал

[Владимир Куц]

out = ''
создать пустую строку, в которой будем накапливать результат
for im in str(number):
преобразуем число в строку и итерируем по одному символу числа
out += str(int(im)**2)
полученный символ - цифру числа преобразуем в целое число и возводим в квадрат, затем преобразуем снова в строку и конкатенируем с переменной в которой накапливаем результирующую строку

а так разве можно?:)

да - так можно
return int(out)
результирующую строку преобразуем снова в число и возвращаем результат

Answer 2

[Andrey Dugin]

Есть большое количество способов это сделать:

def square_digits_v1(n):
    return int(''.join(str(int(x) ** 2) for x in str(n)))

def square_digits_v2(n, t=str.maketrans({str(d): str(d ** 2) for d in range(10)})):
    return int(str(n).translate(t))

def square_digits_v3(n):  # import re
    return int(re.sub('\d', lambda m: str(int(m.group(0)) ** 2), str(n)))

def square_digits_v4(n):  # import numpy as np
    return int(''.join((np.fromiter(str(n), int) ** 2).astype(str)))

def square_digits_v5(n):  # Только вывод на печать
    print(*[int(x) ** 2 for x in str(n)], sep='')

UPD Извращаться, так извращаться:

from decimal import Decimal
from io import StringIO

def square_digits_v6(n):
    with StringIO() as buffer:
        for digit in Decimal(n).as_tuple().digits:
            print(digit ** 2, end='', file=buffer)
        return int(buffer.getvalue())

as_int = lambda f: lambda x: int(f(x))

@as_int
def square_digits_v7(n):
    return eval('f"{'+'**2}{'.join(f'{n}')+'**2}"')

Comments

[Евгений Сноу]

Спасибо огромное я каждую конструкцию разберу для себя!

def square_digits_v5(n): # Только вывод на печать
print(*[int(x) ** 2 for x in str(n)], sep='')
а что означает * после print? я такую вижу впервые конструкцию

[Andrey Dugin]

Евгений Сноу, почитайте про передачу аргументов в функции Python, в частности про *args и **kwargs. Вот, например, поясняющее видео.

[Евгений Сноу]

Andrey Dugin, спасибо!