python的get_polygonal_numbers

Question

测试1：
输入：print(f"First 5 triangular numbers are: {get_polygonal_numbers(3, 5)}")
输出：First 5 triangular numbers are: [1, 3, 6, 10, 15]
测试2：
输入：print(f"First 6 square numbers are: {get_polygonal_numbers(4, 6)}")
输出：First 6 square numbers are: [1, 4, 9, 16, 25, 36]

def get_polygonal_numbers(sides, terms):

Answer 1

def get_polygonal_numbers(s, n):
    temp = []
    for i in range(1, n+1):
        data = ((s-2)*i**2-(s-4)*i)/2
        temp.append(int(data))
    return temp


print(get_polygonal_numbers(4, 6))