3d数组中每n个切片的平均值- python

import numpy as np

def average_slices(arr, n):
    # 获取数组的形状
    shape = arr.shape
    
    # 计算切片的数量
    num_slices = shape[0] // n
    
    # 创建一个存储平均值的数组
    averages = np.zeros((num_slices, shape[1], shape[2]))
    
    # 遍历每个切片
    for i in range(num_slices):
        # 计算切片的起始和结束索引
        start = i * n
        end = start + n
        
        # 获取切片并计算平均值
        slice = arr[start:end, :, :]
        average = np.mean(slice, axis=0)
        
        # 将平均值存储到结果数组中
        averages[i, :, :] = average
    
    return averages