Programa que genera números poligonales dependiendo qué patrón, con interfaz.

poligonal.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-

'''
Programa que a través de un menú permite generar 
el n-simo número poligonal (dado por ti) para 
los patrones triangular, cuadrado, pentagonal,
hexagonal. El programa te pregunta si parar
o continuar.

[sobre los números poligonales](https://es.wikipedia.org/wiki/N%C3%BAmero_poligonal)

Lo podéis probar en: https://repl.it/languages/python3
'''

print('''Número poligonal, pues.

1. Generar para un patrón triangulo.
2. Generar para un patrón cuadrado.
3. Generar para un patrón pentágonal.
4. Generar para un patrón hexagonal.
''')
opcion = input("Escoja una opción (número): ")
if not opcion in ["1","2","3","4"] or not opcion.isdigit():
	print('Mala opción. chao.')
	exit()

# nada, necesitamos un entero, pues.
figura = int(opcion)

# función recursiva que genera números poligonales, pues.
def num_poligonal(figura, n):
	factor = (n/2)*(figura*n - (figura-2))

	print(factor)

	otra = input("¿continuar? (y or n): ")
	if otra.lower() == 'y':
		num_poligonal(figura, n+1)
	else:
		return None
# fin de la función, por si os confundís

# vamos a probar.
num_poligonal(figura, 1)