python - 在python中改组多维numpy数组

Question

我有一个多维 NumPy 数组，它充当我的神经网络的输入数据。我有 2115 个不同的 (6,100,60) 样本。我想对 2115 (6,100,60) 个样本的数据进行洗牌 - 重新排列顺序。我想确保 (6,100,60) 的通道、列和行都保持在原位 - 即，我只想打乱这 2115 个样本的位置（在数组中）。

我不知道如何做到这一点 - 任何帮助将不胜感激。

谢谢你。

标签： pythonarraysnumpymachine-learning

Answer 1

您可以打乱样本编号（假设第一维）然后重新索引：

order = np.random.permutation(np.arange(2115))

data = data[order]