python - Numpy 随机函数为给定参数创建不一致的形状

Question

我发现 numpy 的 random number generators 有一个奇怪的行为。 似乎它们不会为给定的参数生成一致的矩阵形状。 之后花费额外的一行进行转换真是太烦人了，我想避免这种情况。 如何直接告诉 matlib.randn 生成大小为 (200,) 的向量？

import numpy as np
A = np.zeros((200,))
B = np.matlib.randn((200,))
print(A.shape)    # prints (200,)
print(B.shape)    # prints (1, 200)

Answer 1

使用numpy.random而不是numpy.matlib:

numpy.random.randn(200).shape # prints (200,)

numpy.random.randn可以创建任何形状，而 numpy.matlib.randn总是创建一个矩阵。