我想生成一个有序数组,然后将它乘以另一个数组:
[ [0,1,2,3,4,5] [0,1,2,3,4,5] [0,1,2,3,4,5] ... [0,1,2,3,4,5] ]
我可以用 nums = np.arange(0, 6) 生成第一个 [0,1,2,3,4,5] 但是如果我乘以列表中的数字它只是增加值 = [nums* 3] = [0,3,6,9,12,15]。 我该怎么做?
请您参考如下方法:
使用 numpy 方法(numpy.repeat 和 numpy.expand_dims):
np.repeat(np.expand_dims(np.arange(0,6), axis=0), repeats=5, axis=0)
array([[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5]])
或者,更简单地说:
np.repeat([np.arange(0,6)],repeats=5, axis=0)
如果您尝试扩展预先存在的一维数组,第一种方法很有用。如果您尝试从头开始创建数组,则第二种方法更为直接。
