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

Vẽ điểm phân tán 3D với kích thước marker thay đổi trong Matplotlib

Để vẽ scatter 3D với kích thước marker thay đổi, ta thực hiện các bước sau:

Các bước thực hiện

  • Thiết lập kích thước hình và khoảng đệm.
  • Tạo xs, ys, zs bằng NumPy.
  • Khởi tạo biến s cho kích thước marker biến đổi.
  • Tạo hình bằng figure().
  • Thêm axes 3D bằng add_subplot().
  • Vẽ scatter với scatter(), truyền s và colormap.
  • Hiển thị 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

xs = np.random.randint(low=8, high=30, size=35)
ys = np.random.randint(130, 195, 35)
zs = np.random.randint(30, 160, 35)

s = zs / ((ys * 0.01) ** 2)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(ys, zs, xs, s=s * 5, c=xs, cmap='copper')
plt.show()

Kết quả

Vẽ điểm phân tán 3D với kích thước marker thay đổi trong Matplotlib