Налаштуйте `simplify` у Sympy, щоб дати вихід у потрібному форматі, з повноваженнями 2 комбіновані - sympy

Я хочу спростити вираз за допомогою simplify функція в sympy, і потрібно вивести у спеціальному форматі

simplify("2*((2**n)+1)**2").expand(Basic=True)

Однак sympy повертає мені наступний результат:

2*2**(2*n) + 4*2**n + 2

Але я хочу результат у такому форматі:

2**(2*n+1) + 2**(n+2) + 2

Чи є спосіб зробити це?

Відповіді:

SymPy "s спростити Модуль пропонує кілька методів спрощення: тут вам потрібен один powsimp.

from sympy import *
var("n")
a = simplify(2*((2**n)+1)**2).expand(Basic=True)
powsimp(a)

вихід: 2**(n + 2) + 2**(2*n + 1) + 2