我如何随机地获取numpy数组中一定数量的元素，并且每个类至少有一个元素？

import numpy as np

# 假设你有一个numpy数组arr
arr = np.array([1, 2, 3, 1, 2, 3, 4, 5, 6])

# 获取数组中的唯一值
unique_values = np.unique(arr)

# 初始化结果数组
result = []

# 对每个唯一值进行遍历
for value in unique_values:
    # 获取当前唯一值的索引
    indices = np.where(arr == value)[0]
    # 随机选择一个索引，并将对应元素添加到结果数组
    random_index = np.random.choice(indices)
    result.append(arr[random_index])

# 将结果数组转换为numpy数组
result = np.array(result)

# 输出结果
print(result)