Pyplot：Cmap（索引）->顏色函數

Question

假設我有以下內容：

data_mats = [ ... ]

# pseudo code
cmap = plt.(...).get_cmap('magma', low=0, high=len(data_mats))

for i in range(0,len(data_mats)):

    # pseudo code
    color = cmap(i)

    mat = data_mats[i]
    plt.plot(mat[:,1],mat[:,2], color=color)

如何填寫偽代碼行：

# lines of pseudo code
cmap = plt.(...).get_cmap('magma', low=0, high=len(data_mats))

color = cmap(i)

使用pyplot嗎？

（為什么所有在matplotlib網站上的密集示例，而不是某些代碼規范？）

Answer 1

import matplotlib.pyplot as plt

i = 10
file_ct = 100

cmap = plt.cm.magma
color = cmap(float(i+1) / float(file_ct))