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

Vẽ scatter mảng numpy 2D trong matplotlib


Để vẽ scatter mảng numpy 2D trong matplotlib, thực hiện các bước sau:

Các bước

  • Thiết lập kích thước hình và padding.
  • Tạo dữ liệu ngẫu nhiên kích thước 100x3.
  • Dùng scatter() để vẽ mảng 2D (cột 0, 1 làm x, y; cột 2 làm màu).
  • Hiển thị hình bằng show().

Ví dụ

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.array(np.random.random((100, 3)))
plt.scatter(data[:, 0], data[:, 1], c=data[:, 2], cmap='hot')
plt.show()

Kết quả

Biểu đồ.scatter 100 điểm ngẫu nhiên, màu theo cột thứ 3 theo colormap hot.