迭代python列表并求和相等的项

def find_subarrays_with_sum(arr, target_sum):
    subarrays = []
    current_sum = 0
    prefix_sums = {0: [-1]}  # 初始化前缀和字典，用于记录前缀和及其对应的索引

    for i, num in enumerate(arr):
        current_sum += num
        if (current_sum - target_sum) in prefix_sums:
            for start_index in prefix_sums[current_sum - target_sum]:
                subarrays.append(arr[start_index + 1:i + 1])
        if current_sum not in prefix_sums:
            prefix_sums[current_sum] = []
        prefix_sums[current_sum].append(i)

    return subarrays

# 示例使用
arr = [1, 2, 3, 4, 5]
target_sum = 5
print(find_subarrays_with_sum(arr, target_sum))