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

Làm cách nào để tạo huyền thoại cho thanh 3D trong Matplotlib?

Để tạo chú giải cho thanh 3D trong matplotlib, chúng ta có thể vẽ các thanh 3D và đặt chú giải bằng cách sử dụng phương thức chú giải ().

Các bước

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
  • Tạo một hình mới hoặc kích hoạt một hình không tồn tại bằng cách sử dụng figure () phương pháp.
  • Thêm các trục vào hình như một phần của sắp xếp ô phụ.
  • Tạo danh sách dữ liệu x3, y3, z3, dx, dy và dz bằng cách sử dụng numpy.
  • Vẽ một thanh 3D bằng cách sử dụng bar3d () phương pháp.
  • Tạo trục hình chữ nhật để đặt chú giải.
  • Sử dụng chú giải () phương pháp để đặt chú giải cho các thanh.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

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
fig = plt.figure()
ax1 = fig.add_subplot(111, projection='3d')
x3 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
y3 = [5, 6, 7, 8, 2, 5, 6, 3, 7, 2]
z3 = np.zeros(10)
dx = np.ones(10)
dy = np.ones(10)
dz = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
ax1.bar3d(x3, y3, z3, dx, dy, dz, color="green")
b1 = plt.Rectangle((0, 0), 1, 1, fc="green")
ax1.bar3d(y3, x3, z3, dx, dy, dz, color="red")
b2 = plt.Rectangle((0, 0), 1, 1, fc="red")
ax1.legend([b1, b2], ['type1', 'type2'])
plt.show()

Đầu ra

Làm cách nào để tạo huyền thoại cho thanh 3D trong Matplotlib?