bennylope

#!/usr/bin/env python

"""
A script for parsing a flake8 error log and generating useful stats about the
errors in the code.

Author: Ben Lopatin (I think I wrote it, at least, no guarantee)
License: BSD
"""

ken333135

def genSankey(df,cat_cols=[],value_cols='',title='Sankey Diagram'):
    # maximum of 6 value cols -> 6 colors
    colorPalette = ['#4B8BBE','#306998','#FFE873','#FFD43B','#646464']
    labelList = []
    colorNumList = []
    for catCol in cat_cols:
        labelListTemp =  list(set(df[catCol].values))
        colorNumList.append(len(labelListTemp))
        labelList = labelList + labelListTemp