关于在python 3.6中编写以下问题代码的任何想法或线索吗?
想象一下,我的矩阵A和B如下: A = [1 2,3 4,5 6],尺寸为3 * 2 B = [1、3、5],尺寸为3 * 1
现在,我想使用索引随机更改行。
例如,索引1与A的[1 2]和B的[1]相关,索引2与A的[3 4]和B的[3]相关,索引3与A的[5 6]的相关以及[5]来自B。
想象一下,我随机将索引排序为2、3、1,现在我的输出将是: A = [3 4,5 6,1 2] B = [3,5,1]
答案 0 :(得分:1)
import numpy as np
A = [[1, 2],[3, 4], [5, 6]]
A = np.array(A)
B = [[1], [3], [5]]
B = np.array(B)
import random
def rand(n):
l = list(range(n))
random.shuffle(l)
l = np.reshape(l, (n,1)) return l l = rand(3)
print(l)
AF = []
AFF = []
BF = []
BFF = []
for i in range (0, len(A)):
AF = A[l[i]]
AFF.extend(AF)
BF = B[l[i]]
BFF.extend(BF)
B = np.array(BFF)
A = np.array(AFF)
print(B)
print(A)