获取 numpy 的正方形和立方体并附加它

Question

我有一个像下面这样的 numpy 数组：-

array([[2],
       [3],
       [4],
       [5]])

我想对每个元素进行平方和立方，然后将其附加到 numpy 中。我想要如下所示的输出

array([[2, 4, 8],
       [3, 9, 27],
       [4, 16, 64],
       [5, 25, 125]])

Answer 1

使用 np.power 和 numpy broadcasting 属性 -

np.power(a, [1,2,3])

#array([[  2,   4,   8],
#       [  3,   9,  27],
#       [  4,  16,  64],
#       [  5,  25, 125]], dtype=int32)

getting square and cube of numpy and appending it