将colormap作为参数传递给函数

抱歉，如果此问题已在其他地方解决。在此站点上有关将对象传递给函数的搜索导致与c ++相关的帖子。Google结果指向与生成不同颜色图有关的条目。

背景

我想使用matplotlib颜色图为线图着色。此外，我希望能够指定哪个颜色图，就像我所说的functrion一样。

方法：

import numpy as np, matplotlib.pyplot as plt

data = [[1,2,3,4], [5,6,7,8], [6,7,83,5]]

def plotting_function(data, color_map=plt.cm.viridis, plot_kwds=None)

    colors = color_map(np.linspace(0,1,len(data))

    if plot_kwds is None:
        plt_kwds = {}

    for i, y in enumerate(data):
        x_axis = range(len(y))
        plot_kwds['color'] = colors[i]
        plt.plot(x_axis, y, **plot_kwds)

预期：

一个根据绿线着色的线条图。

其实：

线条颜色不符合虹膜的图形。

我意识到我可以通过colors = plt.cm.viridis(np.linspace(0,1,len(data))移入该函数来解决此问题，但是每次我想更改颜色时都弄乱源代码会很不方便。

题：

如何将颜色图传递给自定义函数？我缺少一种更简单的方法吗？

我非常感谢您的帮助！