python的get_polygonal_numbers

测试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):

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))

公式都给出了，你代入就可以了啊。

您好，我是有问必答小助手，您的问题已经有小伙伴帮您解答，感谢您对有问必答的支持与关注！
PS：问答VIP年卡 【限时加赠：IT技术图书免费领】，了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632