import pandas as pd
import plotly.graph_objects as go


def heatmap(table, file_output=None, title=None, autosize=True, width=800, height=1000):
    """
    Args:
        table:
        file_output:
        title:
        autosize:
        width:
        height:
    """
    if type(table.columns) == pd.MultiIndex:
        columns = table.columns.to_series().apply(lambda x: '{0}-{1}'.format(*x))
    else:
        columns = table.columns
    fig = go.Figure(data=go.Heatmap(
        z=table,
        x=columns,
        y=table.index,
        hoverongaps=False, ))
    fig.update_layout(
        title=title,
        autosize=autosize,
        width=width,
        height=height,
    )
    if file_output:
        fig.write_image(file_output)

    return fig