Add waterfall function to sort genes and samples

[hubentu]

Hi,

I think waterfall function could be a usefull feature to add.

Ref:
https://bioconductor.org/packages/devel/bioc/vignettes/maftools/inst/doc/oncoplots.html

from comut import comut
from comut import fileparsers
import pandas as pd

data = pd.read_csv('tutorial_data/tutorial_mutation_data.tsv', sep = '\t')
toy_comut = comut.CoMut()

sample_order, gene_order = toy_comut.waterfall(data)
toy_comut.samples = sample_order
toy_comut.add_categorical_data(data, name = 'Mutation type', category_order = gene_order)

toy_comut.plot_comut(figsize = (18,7))
toy_comut.add_unified_legend()
toy_comut.figure.savefig("comut_waterfall.png")