Loops Python添加基于正方形的金字塔层

我试图创建一个接受用户输入的程序，在本例中为“I”。然后，它使用“i”来计算正方形金字塔中的层数。之后，它将计算具有“i”层的金字塔中的块总数

图层增加如下：

• 第1-1层
• 第2-9层
• 第3-25层
• 第4-49层
• 第5-81层
• 第6-121层

等等

我创建了一个代码，它计算每层的块数，但不加总块数。例如，如果我输入'3'，它将输出1,9,25，但我需要它输出35，这是这些数字的总和

如果您有任何帮助，我们将不胜感激。这是我的代码

while True:
    i = input('Enter a number: ')
    i = int(i) #2
    while i >= 1: #2  
        i2 = i #2
        a = i - 1 #1
        b = a + i2 #3
        c = b**2   #9
        i = i - 1 #1
        print(c)

---

在第二个while循环上添加count，然后在count中添加c，并在while循环后添加printed count

count = 0
    while i >= 1: #2  
        i2 = i #2
        a = i - 1 #1
        b = a + i2 #3
        c = b**2   #9
        i = i - 1 #1
        count = count +c
    print(count)

---

也许我不明白你的问题，但你只是想在内部while循环上面创建一个变量（可能称为count？），然后在每次迭代中向其添加c，然后在循环后打印它吗？非常感谢，它工作得非常好！还有这么简单的东西！