Computer >> Máy Tính >  >> Lập trình >> Python

Làm cách nào để chuyển đổi các số sang thang màu trong Matplotlib?

Để chuyển đổi các số sang thang màu trong matplotlib, chúng ta có thể thực hiện các bước sau.

Các bước

  • Tạo các điểm dữ liệu x, y và c bằng cách sử dụng numpy.
  • Chuyển đổi các điểm dữ liệu thành khung dữ liệu Pandas.
  • Tạo một hình mới hoặc kích hoạt một hình hiện có bằng phương pháp subplots ().
  • Nhận bản đồ màu nóng.
  • Để chuẩn hóa tuyến tính dữ liệu, chúng tôi có thể sử dụng Normalize () lớp học.
  • Vẽ biểu đồ các điểm phân tán với các điểm dữ liệu x và y và bản đồ màu chuẩn hóa tuyến tính.
  • Đặt xticks cho x điểm dữ liệu.
  • Để tạo thanh màu, hãy tạo một đối tượng có thể ánh xạ vô hướng.
  • Sử dụng colorbar () phương pháp tạo thanh màu.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

 from matplotlib import pyplot as plt, Colorsimport numpy as npimport pandas as 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, color =cmap (chuẩn (df.c.values))) ax.set_xticks (df.x) sm =plt.cm.ScalarMappable (cmap =cmap, chuẩn =chuẩn) fig.colorbar (sm) plt.show ()  

Đầu ra

Làm cách nào để chuyển đổi các số sang thang màu trong Matplotlib?