Plotly Python: Arrange subplots in a grid layout

grid_layout_subplots.py

def grid_layout_subplots(traces: typing.List[go.Trace], titles: typing.List[str] = None, ncolumns: int = 1, fig: go.Figure = None) -> go.Figure:
    if fig is None:
        fig = go.Figure()

    ncolumns = int(min(ncolumns, len(traces)))

    if ncolumns == 0:
        return fig

    nrows = int(math.ceil(len(traces) / ncolumns))

    fig.set_subplots(nrows, ncolumns, subplot_titles=titles)

    for idx, t in enumerate(traces):
        col = int((idx % ncolumns))+1
        row = int(idx / ncolumns)+1

        # if nested list: plot them in a shared axis
        if type(t) is list:
            for t_ in t:
                fig.add_trace(t_, row=row, col=col)
        else:
            fig.add_trace(t, row=row, col=col)

    return fig