Matplotlib中怎么設(shè)置圖例的漸變顏色

小億
159
2024-05-11 19:34:56

要在Matplotlib中設(shè)置圖例的漸變顏色，可以使用LinearSegmentedColormap來創(chuàng)建自定義的顏色映射，然后將這個(gè)顏色映射應(yīng)用到圖例中。

以下是一個(gè)示例代碼，演示如何設(shè)置圖例的漸變顏色：

import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap

# 創(chuàng)建自定義的顏色映射
colors = [(0, 'blue'), (0.5, 'green'), (1, 'red')]
cmap = LinearSegmentedColormap.from_list('my_cmap', colors)

# 創(chuàng)建示例數(shù)據(jù)
x = [1, 2, 3, 4, 5]
y = [1, 4, 9, 16, 25]

# 繪制折線圖
plt.plot(x, y, color='black', label='Data')

# 創(chuàng)建圖例，并設(shè)置漸變顏色
plt.legend(loc='upper left', fancybox=True)
plt.legend().get_frame().set_facecolor('0.9')
plt.legend().legendHandles[0].set_color(cmap(0.5))

plt.show()

在這個(gè)示例中，我們通過LinearSegmentedColormap創(chuàng)建了一個(gè)包含藍(lán)色、綠色和紅色漸變顏色的自定義顏色映射，并將這個(gè)顏色映射應(yīng)用到圖例中。您可以根據(jù)需要調(diào)整colors列表中的顏色和位置值來設(shè)置不同的漸變顏色。

0 踩