随机数生成的python代码

Question

现在需要用python 生成5组随机数向量，每组都是400行乘以1列（其实就是一个400X5）的矩阵。每一列都保证服从正态分布，范围在（0，5] 之间，左开右闭。生成完毕以后，给这个矩阵前面额外加一列，以20行一组进行标记从1开始标记，总共标记数1-20.请给出python代码。

Answer 1

import numpy as np

# 生成5组随机数向量
vecs = np.random.normal(2.5, 1.5, size=(400, 5))
vecs[vecs <= 0] = 0.01
vecs[vecs > 5] = 5

# 构造标记列
markers = np.repeat(np.arange(1, 21), 20).reshape(-1, 1)

# 在矩阵前面额外加一列
matrix = np.hstack((markers, vecs))

print(matrix)