Python中对矩阵的洗牌操作

【code】

import numpy as np

# 创建随机交换的索引
permutation = list(np.random.permutation(3))

# 创建矩阵X,Y
X = np.array([[0, 1, 2], [0, 1, 2], [0, 1, 2]])
Y = np.array([[0, 1, 2]])

# 交换顺序
shuffled_X = X[:, permutation]
shuffled_Y = Y[:, permutation]

# 输出
print("permutation:")
print(permutation)

print("X:")
print(X)
print("Y:")
print(Y)

print("shuffled_X:")
print(shuffled_X)
print("shuffled_Y:")
print(shuffled_Y)

【result】

permutation:
[1, 2, 0]
X:
[[0 1 2]
 [0 1 2]
 [0 1 2]]

Y:
[[0 1 2]]

shuffled_X:
[[1 2 0]
 [1 2 0]
 [1 2 0]]

shuffled_Y:
[[1 2 0]]