from matplotlib.pyplot import get_cmap
import numpy as np
numofLines = 5
cMapType = 'viridis'
colorMap = get_cmap(cMapType) # get_cmap is matplotlib object
colorList = np.linspace(0, 1, numofLines)
lineColors = colorMap(colorList)
# convert to RGB*255
lineColors = lineColors * 255
lineColors = lineColors.astype(int)
print lineColors
# ## Set a custom color map
# colors = [
# (0, 0, 0),
# (45, 5, 61),
# (84, 42, 55),
# (150, 87, 60),
# (208, 171, 141),
# (255, 255, 255)
# ]
# cmap = pg.ColorMap(pos=np.linspace(0.0, 1.0, 6), color=colors)
# imv.setColorMap(cmap)