분산 형 플롯에 대한 컬러 바를 따로 저장

Question

PNG 이미지로 저장하는 색상 막대가있는 산점도가 있습니다. 나는 어떤 그림 크기의 음모가 필요하지만 칼라 바를 추가하는 것은 원래 음모를 조정합니다.

import pylab as plt 

plt.figure(figsize=FIGSIZE) 
plt.scatter(X, Y, c=Z, s=marker_size, norm=LogNorm(), vmin=VMIN, vmax=VMAX, cmap=CMAP,rasterized=True,lw=0,) 
CB = plt.colorbar(ticks=TICKS, format=FORMAT) 

원래 그림 (위의 그림과 같이 설정)과 컬러 바를 각각 별도의 이미지로 저장할 수 있습니까?

Answer 1

확실한 대답은 "귀하의 컬러 바를 별도로 표시"입니다. 첫 번째 그림이 왜곡되지 않도록 새 그림 창을 만들고 여기에 색상 막대를 그려야합니다. 작은 예 :

import matplotlib.pyplot as plt 
import numpy as np # only for dummy data 

X,Y = np.mgrid[-2:3,-2:3] 
Z = np.random.rand(*X.shape) 
FIGSIZE = (2,3) 

plt.figure(figsize=FIGSIZE) 
mpb = plt.pcolormesh(X,Y,Z,cmap='viridis') 
# plot the original without a colorbar 
plt.savefig('plot_nocbar.png') 

# plot a colorbar into the original to see distortion 
plt.colorbar() 
plt.savefig('plot_withcbar.png') 

# draw a new figure and replot the colorbar there 
fig,ax = plt.subplots(figsize=FIGSIZE) 
plt.colorbar(mpb,ax=ax) 
ax.remove() 
plt.savefig('plot_onlycbar.png') 

# save the same figure with some approximate autocropping 
plt.savefig('plot_onlycbar_tight.png',bbox_inches='tight') 

이 생성 된 다음 네 가지 수치를 고려 (제대로 보려면 클릭) :

첫 번째는를 호출하지 않고 그림의 저장된 버전입니다 colormap. 이것은 괜찮습니다. 이것은 여러분이 보존하기를 원하는 것입니다. 두 번째 그림은 colorbar으로 전화를 걸면 어떤 일이 발생하는지 보여줍니다. 원본 그림에서 약간의 공간이 필요하므로이를 방지해야합니다.

원래 그림 크기 인 plt.subplots을 사용하여 새 그림 (및 축)을 열어야합니다. 이렇게하면 생성 된 색상 막대가 원본 그림에서 그려진 것과 같은 크기가 될 수 있습니다. 위의 설정에서 나는 matplotlib이 컬러 바 자체의 크기를 결정하도록했습니다; 그러나 그 후에는 결과 플롯을 오염시킬 보조 축을 삭제해야합니다. (다른 옵션은 새로운 그림에서 컬러 바의 예상 크기로 수동으로 하나의 축을 만드는 것입니다. 이것은 실현 가능한 과정이 아닌 것으로 생각됩니다.)

이제 세 번째 플롯에서 삭제 된 축 뒤에 남은 빈 공간은 결과 플롯에서 명확하게 볼 수 있습니다 (그러나 색상 막대의 크기는 그에 따라 적절합니다). 포스트 프로덕션에서이 공백을 수동으로 잘라내거나 컬러 바 이미지를 자동 자르기하는 것을 사용할 수 있습니다.

matplotlib 자체가 그림의 대부분을 자르기하는 플롯 버전도 포함되었습니다. bbox_inches='tight' 키워드 인수는 savefig으로 정확히 이루어집니다. 단점은 결과 이미지 파일에 네 번째 이미지에서 볼 수 있듯이 색상 막대 만 포함되어 있지만 결과 색상 막대의 크기가 원본과 약간 다를 수 있습니다. 특정 요구에 따라 가장 편리한 솔루션을 찾기 위해 사용 가능한 방법을 시험해야합니다.