python - 展开并填充 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]] ，利用 broadcasting 执行电源操作与 **[2,3] ，因为我们保留了尺寸，然后将输出堆叠为新列 np.c_ (多种堆叠方式之一)。

因此，实现 -

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

示例输出 -

In [902]: np.c_[a,a[:,[1]]**[2,3]]
Out[902]: 
array([[  1,   3,   9,  27],
       [  1,   2,   4,   8],
       [  1,   5,  25, 125]])