pyplot.scatter : 낮은 -> 높음에서 점의 색을 어떻게 설정합니까?

Question

방금 ​​pyplot.scatter을 발견했으며 조금만 놀았습니다. 나는 색칠 때문에 결과를 해석하는 데 어려움을 겪고 있습니다. 에 낮은 값을 설정하는 방법이 있다면

가

import pyplot as plt 
import numpy as np 

plt.scatter(x=dataX.flat, y=dataY.flat, vmin=np.min(dataMag), 
      vmax=np.max(dataMag), c=dataMag.flat, lw=1, s=1) 
plt.show() 

정말 도움이 될 것입니다 : 예를 들어, 제 3 (A, B) 크기의 데이터 배열을 DATAX, dataY 및 dataMag을 부여 dataMag (예 : 0)을 흰색으로 설정하고 점차 최대 색상 인 검정색 또는 빨간색으로 변경하거나 다른 색상으로 변경합니다. 이것을 할 수있는 방법이 있습니까?

미리 감사드립니다.

Answer 1

나는 당신이 kwarg cmap을 찾고 있다고 생각합니다.

plt.scatter(...,cmap='hot') 

는 the scipy cookbook에서 촬영 :

from pylab import * 
from numpy import outer 
rc('text', usetex=False) 
a=outer(arange(0,1,0.01),ones(10)) 
figure(figsize=(10,5)) 
subplots_adjust(top=0.8,bottom=0.05,left=0.01,right=0.99) 
maps=[m for m in cm.datad if not m.endswith("_r") ] 
maps.sort() 
l=len(maps)+1 
for i, m in enumerate(maps): 
    subplot(1,l,i+1) 
    axis("off") 
    imshow(a,aspect='auto',cmap=get_cmap(m),origin="lower") 
    title(m,rotation=90,fontsize=10) 

잘 설치에 사용할 수있는 모든 컬러지도와 그림을 생성합니다. 마지막에 '_r'이 붙은 이름은 반대 방향의 색상과 함께 동일한지도를 제공합니다.

사용자 정의 색상 맵을 정의 할 수도 있습니다.