получить квадрат и куб 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 Broadcast Свойства -

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

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