从多个矩阵中抽取随机数构造新矩阵

Question

有一个三维array， 由 20个 50 X 7矩阵构成 （a1-a20）。

现在要构造一个20 X 7 的新矩阵 M。

构造规则是从第一个 50 X 7 的矩阵（a1）的每列中随机取一个数。作为新矩阵第一行,
第二个50 X 7 的个 50 X 7 的矩阵（a2）的每列中。作为新矩阵第二行
依次类推取到第20个矩阵(a20)，构成新矩阵的第20行。

这个构造规则如何实现？

Answer 1

import random

import numpy as np

data = np.random.randint(0, 100, size=(20, 50, 7))
result = []
for i in data:
    temp = []
    for j in range(0, len(data[0][0])):
        randint = random.randint(0, len(data[0]) - 1)
        temp.append(i[randint][j])
    result.append(temp)
print(len(result))
print(result)