Fix marker_size indexing for ColormappedScatterPlot
Use itertools to avoid unnecessary array and list creation