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

Tăng chất lượng colormap và linewidth trong streamplot Matplotlib

Để tăng chất lượng colormap và linewidth trong streamplot của Matplotlib, bạn có thể thực hiện theo các bước sau:

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

  • Thiết lập kích thước hình vẽ và điều chỉnh khoảng cách giữa các subplot.
  • Tạo figure và tập hợp subplot.
  • Tạo điểm dữ liệu x và y, sau đó dùng np.meshgrid() để tạo ma trận tọa độ.
  • Tính toán X và Y từ x và y.
  • Tạo streamplot với dữ liệu x, y, X, Y. Tăng linewidth (ví dụ: 5) và chọn cmap (ví dụ: 'plasma').
  • Tạo colorbar cho instance stream.lines.
  • Sử dụng phương thức show() để hiển thị hình vẽ.

Ví dụ minh họa

import numpy as np
from matplotlib import pyplot as plt

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

fig, ax = plt.subplots()
x, y = np.meshgrid(np.linspace(-5, 5, 20), np.linspace(-5, 5, 20))
X = y
Y = 3 * x - 4 * y

stream = ax.streamplot(x, y, X, Y, density=1, linewidth=5, cmap='plasma', color=Y)
fig.colorbar(stream.lines, ax=ax)
plt.show()

Kết quả đầu ra

Biểu đồ streamplot với đường dày (linewidth=5), colormap 'plasma' và colorbar bên cạnh.