Làm thế nào để tô màu cầu vồng dưới một đường cong trong Python Matplotlib?

Để tô màu cầu vồng dưới một đường cong 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 ô phụ.
• Tạo một phương thức do người dùng xác định, plot_rainbow_under_curve () , có thể có danh sách 7 màu cầu vồng và tạo tập hợp các điểm dữ liệu "x" sử dụng numpy.
• Lặp lại trong phạm vi từ 0 đến 7 và vẽ đường cong và tô vùng giữa đường cong đó.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
def plot_rainbow_under_curve():
   rainbow_colors = ['violet', 'indigo', 'blue', 'green', 'yellow', 'orange', 'red']
   x = np.linspace(-5, 5, 100)
   for i in range(0, 7):
      plt.plot(x, x ** 2, lw=0)
      plt.fill_between(x, x ** 2 + len(rainbow_colors)-i,color=rainbow_colors[i])

plot_rainbow_under_curve()

plt.show()

Đầu ra

Nó sẽ tạo ra kết quả sau