Cách tô màu cầu vồng dưới đường cong trong Python Matplotlib

Giới thiệu

Để tô màu cầu vồng (rainbow) dưới một đường cong trong Python Matplotlib, bạn có thể làm theo các bước:

• Thiết lập kích thước hình vẽ và padding.
• Tạo hàm tự định nghĩa plot_rainbow_under_curve() chứa danh sách 7 màu cầu vồng và tạo dữ liệu x bằng numpy.
• Lặp từ 0 đến 6, vẽ đường cong và tô màu vùng giữa đường cong đó và đường cơ sở bằng fill_between.
• Hiển thị hình vẽ bằng show().

Ví dụ minh họa

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()

Kết quả đầu ra

Hình vẽ đường cong parabol với 7 lớp màu cầu vồng xếp chồng lên nhau dưới đường cong: