Question

嗨，我想创建一个10 x 5的矩阵，第一列填充1到10，随后的列填充第二列的2、3、4和5倍。

我已经使以下代码可以工作，但是有没有更短的方法呢？

import numpy as np
mat = np.zeros([10,5])
mat[:,0] = np.arange(1,11)
mat[:,1] = np.dot(mat[:,0],2)
mat[:,2] = np.dot(mat[:,0],3)
mat[:,3] = np.dot(mat[:,0],4)
mat[:,4] = np.dot(mat[:,0],5)

Answer 1

我认为您可以通过外部产品来实现这一目标。

尝试：

import numpy as np

a = np.arange(1, 11).reshape(-1, 1)    # column vector (1,2,3,...,10)
b = np.arange(1, 6).reshape(1, -1)     # row vector (1,2,3,...,5)
np.matmul(a, b)                        # matrix of entries of multiplication of the indices (1-based indices)

或单线：

np.arange(1, 11).reshape(-1, 1) * np.arange(1, 6).reshape(1, -1)

Answer 2

这对我有用：

>>> np.array([np.array([1,2,3,4,5]) * i for i in range(1,11)])
array([[ 1,  2,  3,  4,  5],
       [ 2,  4,  6,  8, 10],
       [ 3,  6,  9, 12, 15],
       [ 4,  8, 12, 16, 20],
       [ 5, 10, 15, 20, 25],
       [ 6, 12, 18, 24, 30],
       [ 7, 14, 21, 28, 35],
       [ 8, 16, 24, 32, 40],
       [ 9, 18, 27, 36, 45],
       [10, 20, 30, 40, 50]])

Answer 3

这正是内置numpy outer所做的：

>>> np.outer(np.arange(1, 11), np.arange(1, 6))
array([[ 1,  2,  3,  4,  5],
       [ 2,  4,  6,  8, 10],
       [ 3,  6,  9, 12, 15],
       [ 4,  8, 12, 16, 20],
       [ 5, 10, 15, 20, 25],
       [ 6, 12, 18, 24, 30],
       [ 7, 14, 21, 28, 35],
       [ 8, 16, 24, 32, 40],
       [ 9, 18, 27, 36, 45],
       [10, 20, 30, 40, 50]])

编写此numpy代码的更有效方法

3 个答案: