帮我用python写一个数据增强

好的，我可以为您提供一个简单的数据增强的 Python 代码片段，如下所示：

import numpy as np
from scipy import ndimage

def random_rotation(image_array: np.ndarray, angle_range: int) -> np.ndarray:
    """
    Randomly rotate the given image array by an angle within the given range.
    :param image_array: A 3D numpy array representing an image.
    :param angle_range: An integer representing the maximum angle of rotation in degrees.
    :return: A 3D numpy array representing the rotated image.
    """
    random_angle = np.random.uniform(-angle_range, angle_range)
    rotated_array = ndimage.rotate(image_array, random_angle, reshape=False)
    return rotated_array

这是一个简单的数据增强函数，它可以随机旋转给定的图像数组并返回旋转后的数组。您可以根据需要调整此代码来添加不同的数据增强功能。