得到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)