Python 代码：同步洗牌两个数组 (shuffle_in_unison)

def shuffle_in_unison(a, b):
    rng_state = np.random.get_state()
    np.random.shuffle(a)
    np.random.set_state(rng_state)
    np.random.shuffle(b)
    return a, b

代码分析：

1. rng_state = np.random.get_state()：保存当前的随机数生成器状态，以便后续恢复。
2. np.random.shuffle(a)：使用 NumPy 的 shuffle 函数对数组 a 进行洗牌操作，改变了 a 的顺序。
3. np.random.set_state(rng_state)：将随机数生成器的状态恢复为之前保存的状态，确保后续对 b 的洗牌操作使用相同的随机数序列。
4. np.random.shuffle(b)：使用 NumPy 的 shuffle 函数对数组 b 进行洗牌操作，改变了 b 的顺序，并且由于随机数序列相同，保证 b 的顺序与 a 保持一致。
5. return a, b：返回洗牌后的数组 a 和 b。

总结：

这个函数用于以相同的顺序对两个数组进行洗牌操作，保证两个数组的顺序一致。这在需要对数据进行随机处理，同时保持对应关系的情况下非常有用。例如，在训练机器学习模型时，可以对特征和标签进行同步洗牌，避免数据顺序错乱导致的训练结果偏差。