Pyplot: Cmap (index) -> функция цвета

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