用python代码写一个前缀和函数

以下是一个简单的前缀和函数的Python代码：

def prefix_sum(nums):
    """
    计算输入列表的前缀和
    
    :param nums: 输入列表
    :return: 前缀和列表
    """
    prefix = [0] * (len(nums) + 1)
    for i in range(1, len(nums) + 1):
        prefix[i] = prefix[i - 1] + nums[i - 1]
    return prefix[1:]

这个函数接受一个列表作为参数，并返回一个新的列表，其中每个元素都是输入列表中对应位置之前所有元素的和。函数使用一个长度为输入列表长度加一的列表来存储前缀和，然后使用一个循环来计算每个位置的前缀和。最后，函数返回前缀和列表中除第一个元素之外的所有元素，因为第一个元素始终为零。