2D 및 3D numpy 배열의 다형성 처리

Question

저는 파이썬에서 이미지 작업을하고 있으며 컬러 및 회색조 이미지를 모두 처리 할 수있는 프로그램을 작성해야합니다. 그레이 스케일의 (512,512,3)

모양 : : (512512) 이제

나는 이미지의 모든 채널을 통해 루프를 가지고, 즉 반환 :

그들은 NumPy와 배열을 컬러의

모양이야

For Color: im[:,:,0], im[:,:,1], im[:,:,2] 

For Grayscale: im[:,:] 

if 조건없이 같은 형식으로 작성하는 방법은 무엇입니까? 나는 그레이 스케일을 위해 im [:, :, 0]을 시도했지만 인덱스의 범위를 벗어났습니다. im_new[i,j,k] 어떤을위한 im_old[i,j]과 동일하도록 내가 알고있는 것처럼

im_new = im_old[:,:,np.newaxis] 

이 그것을 만드는 :

Answer 1

난이 도움이인지 아닌지 확실하지 않다

하지만 numpy는 new axes 삽입 할 수있는 기능을 제공합니다 k.

(np.newaxis은 None의 별칭 임)