python - 计算范围内所有 n 个数的四面体编号，将其存储在列表中并返回

Question

定义一个接受两个输入的函数 1. startvalue 2. endvalue 计算一个不包括起始值和结束值的四面体。以列表格式存储输出并返回主函数。 将列表转换为 str 以显示 这是我尝试过但无法获得完整输出的方法。

def calculateNTetrahedralNumber(startvalue, endvalue):
    n = startvalue
    y = endvalue+1
    result = []
    while (n < y):
        result.append(((n*(n+1)*(n+2))/6))
        n = n+1
    return(result)
        

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    startvalue = int(input().strip())

    endvalue = int(input().strip())

    result = calculateNTetrahedralNumber(startvalue, endvalue)
    print(result)

    fptr.write('\n'.join(map(str, result)))
    fptr.write('\n')

    fptr.close()

用户输入的数据 起始值 = 1 最终值(value) 7 输出 1

预期输出

1 4个 10 20 35 56 84

Answer 1

试试这个:

def tetra_n(startvalue, endvalue):
    n = startvalue
    y = endvalue+1
    result = []
    while (n < y):
        result.append(int((n*(n+1)*(n+2))/6))
        n = n+1
        return(result)