홈>2020-02-26 10:19

python - numpy의 정사각형과 입방체를 가져 와서 추가

아래처럼 numpy 배열이 있습니다 :-

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

각 요소를 제곱하고 정육면체를 만들고 numpy에 추가하고 싶습니다. 아래 출력을 원합니다

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

답변 # 1

np.power 사용 numpy 방송 속성 사용-

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

관련 질문