Question

我想生成一个有序数字数组，然后将其乘以另一个数组：

[ [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]。我该怎么办？

Answer 1

使用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)

如果您尝试扩展预先存在的一维数组，则第一种方法很有用。如果您想从头开始创建数组，那么第二种方法会更简单。

Answer 2

顺便说一句，为什么不像下面这样简单地使用SELECT P.NAME, P.CITY, A.NAME, A.CITY, A.STATE FROM PUBLISHER P, AUTHOR A WHERE A.STATE = P.STATE AND A.CITY = "Chicago" AND A.STATE = "IL" ORDER BY P.NAME, P.CITY, A.NAME, A.CITY, A.STATE：

np.array()

单行：

In [147]: nums = np.arange(6)

In [148]: nums
Out[148]: array([0, 1, 2, 3, 4, 5])

In [149]: [nums] * 5
Out[149]: 
[array([0, 1, 2, 3, 4, 5]),
 array([0, 1, 2, 3, 4, 5]),
 array([0, 1, 2, 3, 4, 5]),
 array([0, 1, 2, 3, 4, 5]),
 array([0, 1, 2, 3, 4, 5])]

In [150]: np.array([nums] * 5)
Out[150]: 
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]])

Answer 3

您不能将numpy数组与标量相乘，并且期望与将python列表（或字符串）与标量相乘具有相同的行为。

对于numpy，乘法运算符将在所有数组元素上广播乘法：

即

np.array([1,2,3]) * 2 == np.array([1*2, 2*2, 3*2) == np.array([2,4,6])

相反，您可以使用列表理解

np.array([np.arange(6) for _ in range(4)])
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]])

或通过乘法生成列表列表，然后转换为numpy数组并调整形状：

np.array([list(range(6))*4]).reshape(4,6)
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]])

或生成形状为（1,6）的数组，并沿第一个轴重复：

np.repeat(np.arange(6).reshape(1,6), repeats=4, axis=0)
# produces the same output as the example outputs above.

Answer 4

如果要对数组重复X次，简单的方法是：

X = 3
[list(range(6)) for i in range(X)]

输出：

[[0, 1, 2, 3, 4, 5], [0, 1, 2, 3, 4, 5], [0, 1, 2, 3, 4, 5]]

numpy-创建数组副本的多维数组

4 个答案: