使用python多处理通过迭代返回多个变量的平均值

import multiprocessing

def calculate_average(values):
    total_sum = 0
    count = 0
    for value in values:
        total_sum += value
        count += 1
    average = total_sum / count
    return average

def process_task(values, output):
    average = calculate_average(values)
    output.put(average)

output = multiprocessing.Queue()

data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
num_processes = 4
chunk_size = len(data) // num_processes

processes = []
for i in range(num_processes):
    start = i * chunk_size
    end = start + chunk_size if i < num_processes - 1 else len(data)
    subset = data[start:end]
    
    process = multiprocessing.Process(target=process_task, args=(subset, output))
    processes.append(process)
    process.start()

for process in processes:
    process.join()

total_sum = 0
count = 0
while not output.empty():
    average = output.get()
    total_sum += average
    count += 1

overall_average = total_sum / count

print("Overall Average:", overall_average)