matplotlib兩種顏色之間的顏色漸變

Question

我想要matplotlib中黑色和紅色之間的顏色漸變，其中低值是黑色，並且隨着Y值的增加而變得越來越紅。

import matplotlib.pyplot as plt
xvals = np.arange(0, 1, 0.01)
yvals = xvals
plt.plot(xvals, yvals, "r")
axes = plt.axes()
plt.show()

為了獲得這樣的顏色漸變，我必須更改什么？

Answer 1

您可以從matplotlib文檔中查看此鏈接作為示例。

要創建該顏色圖，您只需要執行以下操作：

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

colors = [(0, 0, 0), (1, 0, 0)] # first color is black, last is red
cm = LinearSegmentedColormap.from_list(
        "Custom", colors, N=20)
mat = np.indices((10,10))[1]
plt.imshow(mat, cmap=cm)
plt.show()

結果是：