展开并填充numpy数组

Question

扩展和填充python numpy数组的最简单方法是什么？最初的numpy数组是：

  

[[1,3]，[1,2]，[1,5]]

扩展后，数组应为：

  

[[1,3,9,27]，[1,2,4,8]，[1,5,25,125]]

新添加的列是第二列的平方和立方值。

谢谢！

Answer 1

使用a[:,[1]]切割第二列保持维度，使用**[2,3]执行权限操作，因为我们保留了维度，然后将输出作为新列堆叠{ {1}}（堆叠的众多方法之一）。

因此，实施 -

np.c_

示例输出 -

np.c_[a,a[:,[1]]**[2,3]]

Answer 2

您只需跳过每行1，然后使用广播形成所需的笛卡尔积。

import numpy as np

a = np.array([[1,3], [1,2], [1,5]])
b = a[:, 1:] ** np.arange(4)
print(b)

<强>输出

[[  1   3   9  27]
 [  1   2   4   8]
 [  1   5  25 125]]