Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Matplotlib에서 숫자를 색상 스케일로 변환하려면 어떻게 해야 합니까?

<시간/>

matplotlib에서 숫자를 색상 스케일로 변환하려면 다음 단계를 수행할 수 있습니다.

단계

  • numpy를 사용하여 x, y 및 c 데이터 포인트를 생성합니다.
  • 데이터 포인트를 Pandas 데이터 프레임으로 변환합니다.
  • subplots() 메소드를 사용하여 새 Figure를 생성하거나 기존 Figure를 활성화합니다.
  • 핫 컬러맵을 가져옵니다.
  • 데이터를 선형으로 정규화하려면 Normalize()를 사용할 수 있습니다. 수업.
  • x 및 y 데이터 포인트와 선형 정규화된 컬러맵으로 산점도를 플로팅합니다.
  • xticks 설정 x 데이터 포인트에 대해.
  • 컬러바를 만들려면 스칼라 매핑 가능한 개체를 만드세요.
  • colorbar() 사용 컬러바를 만드는 방법입니다.
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

 matplotlib에서 pyplot을 plt로 가져오기, numpy를 npimport로 pandas를 pdplt.rcParams["Figure.figsize"] =[7.00, 3.50]plt.rcParams["Figure.autolayout"] =Truex =np.arange(12)로 가져오기 )y =np.random.rand(len(x)) * 20c =np.random.rand(len(x)) * 3 + 1.5df =pd.DataFrame({"x":x, "y":y , "c":c})fig, ax =plt.subplots()cmap =plt.cm.hotnorm =colors.Normalize(vmin=2.0, vmax=5.0)ax.scatter(df.x, df.y, 색상 =cmap(norm(df.c.values)))ax.set_xticks(df.x)sm =plt.cm.ScalarMappable(cmap=cmap, norm=norm)fig.colorbar(sm)plt.show() 

출력

Matplotlib에서 숫자를 색상 스케일로 변환하려면 어떻게 해야 합니까?