python - 如何在 python 中随机洗牌数据和目标？

Question

我有一个 4D 数组训练图像，其维度对应于 (image_number,channels,width,height)。我还有一个 2D 目标标签，其尺寸对应于 (image_number,class_number)。训练时，我想使用 random.shuffle 随机打乱数据，但如何保持标签按图像的相同顺序打乱？谢谢!

Answer 1

from sklearn.utils import shuffle
import numpy as np

X = np.array([[0, 0, 0], [1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]])
y = np.array([0, 1, 2, 3, 4])
X, y = shuffle(X, y)
print(X)
print(y)



[[1 1 1]
 [3 3 3]
 [0 0 0]
 [2 2 2]
 [4 4 4]] 

[1 3 0 2 4]