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

Vẽ một đường nhiều màu dựa trên một điều kiện trong Python Matplotlib

Để vẽ một đường nhiều màu dựa trên một điều kiện trong Python Matplotlib, chúng ta có thể thực hiện các bước sau -

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô con.

  • Tạo y điểm dữ liệu bằng cách sử dụng numpy.

  • Làm cho l u điểm dữ liệu để phân biệt màu sắc.

  • Vẽ biểu đồ u l điểm dữ liệu bằng cách sử dụng plot () phương pháp, với các màu sắc khác nhau.

  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import matplotlib.pyplot as plt
import numpy as np

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

y = np.sin(np.linspace(-10, 10, 100))

u = y.copy()
l = y.copy()

u[u <= 0] = np.nan
l[l >= 0] = np.nan

plt.plot(u, color='red')
plt.plot(l, color='blue')

plt.show()

Đầu ra

Vẽ một đường nhiều màu dựa trên một điều kiện trong Python Matplotlib