python定义函数完成前n项和

在Python中，你可以定义一个函数来计算前n项和，通常用于等差数列或等比数列的求和。这里是一个简单的例子，使用循环来计算整数列表的前n项和：

def sum_of_n_terms(n, sequence_type="arithmetic"):
    """
    计算前n项和，支持等差数列（默认）和等比数列。

    参数:
    n (int): 需要计算和的项数。
    sequence_type (str, optional): 序列类型，默认为"arithmetic"表示等差数列。可以设置为"geometric"以处理等比数列。

    返回:
    total (float): 前n项和。
    """
    if sequence_type == "arithmetic":
        # 等差数列求和
        first_term = 0
        common_difference = 1
        total = n * (first_term + (n - 1) * common_difference) / 2
    elif sequence_type == "geometric":
        # 等比数列求和（如果序列有公共除数）
        first_term = 1
        common_ratio = 2  # 这里假设公比为2，替换为实际的值
        total = first_term * (1 - common_ratio**n) / (1 - common_ratio)
    else:
        raise ValueError("sequence_type must be either 'arithmetic' or 'geometric'.")

    return total

# 使用示例
n = 5
print(sum_of_n_terms(n))  # 输出等差数列的前5项和
print(sum_of_n_terms(n, "geometric"))  # 输出等比数列的前5项和