본문 바로가기
Python

plt 축 제거하고 이미지만 저장하기

by saoh 2021. 10. 3.

matplotlib.pyplot.axis('off')명령을 사용해 축을 숨길 수 있다.

#a는 숫자 이미지를 나타내는 배열임
plt.imshow(a.reshape(28,28), cmap='Greys') #이미지 배열을 수정하고, 그레이 스케일로 변환
plt.axis('off') #축 사용하지 않음
plt.savefig(fname='./AA_samples/photo.png', bbox_inches='tight', pad_inches=0) #주변의 패딩 제거

 

다음과 같이 축과 공백이 제거되어 저장된 사진을 확인할 수 있다. 

축과 공백이 제거된 사진들

 

bbox_inches='tight', pad_inches=0 부분을 제거하고 저장하면, c_x_x의 모습으로 저장된다.

저장된 사진들