[python] [numpy] (x, y, z)값을 원소로 갖는 NxMx3 행렬에 대해 (3, 3) 행렬 곱

조회수 1249회

안녕하세요, 이번에도 numpy 상에서 연산을 진행하는데 좀더 쉬운 방법을 찾고자 질문 올려봅니다 ㅎㅎ

이미지

이런 (x, y, z)를 원소로 가지는 3x3x3 numpy 행렬에 대해, (3, 3)의 변환 행렬을 곱해서 3x3x3의 행렬을 얻고 싶습니다.

물론, for문 돌리면 간단히 해결할 수 있는데, 뭔가 numpy 특징을 잘 활용하면 몇줄로 바로 구현할 수 있을 것 같아서요 ㅎㅅㅎ

그냥 단순히 np.matmul로
(R = (3, 3) rotation matrix )

out = np.matmul(R, input[:, :])
#out = np.matmul(R, input) <- 이 녀석도 오류!

로 하니 오류가 뜨더군요....

도움 부탁드립니다!! :)

  • (•́ ✖ •̀)
    알 수 없는 사용자

답변을 하려면 로그인이 필요합니다.

프로그래머스 커뮤니티는 개발자들을 위한 Q&A 서비스입니다. 로그인해야 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)