Để 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