Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Chuyển đổi giá trị dữ liệu thành thông tin màu trong Matplotlib

Để ánh xạ giá trị số liệu thành màu sắc tùy chỉnh, ta lấy colormap bằng plt.get_cmap() và áp dụng lên mảng giá trị.

Các bước

  • Thiết lập figure size và autolayout.
  • Lấy colormap instance: plasma = plt.get_cmap('GnBu_r').
  • Tạo mảng giá trị ngẫu nhiên values.
  • Tạo tọa độ x, y ngẫu nhiên.
  • Vẽ scatter với c=[plasma(val) for val in values].
  • Hiển thị bằng show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

plasma = plt.get_cmap('GnBu_r')
values = np.random.rand(100)
x = np.random.rand(len(values))
y = np.random.rand(len(values))
sc = plt.scatter(x, y, c=[plasma(val) for val in values])
plt.show()

Kết quả

Chuyển đổi giá trị dữ liệu thành thông tin màu trong Matplotlib