Generic scatterplot function for Altair

plot.py

def plot(df, x, y, args, fill=False):
    df = df.reset_index()
    index = df.columns[0]

    if fill:
        chart = alt.Chart(df).mark_circle(size=30)
    else:
        chart = alt.Chart(df).mark_point(size=30)

    chart = chart.encode(
        x=alt.X(x),
        y=alt.Y(y),
        tooltip=[index, x, y, *args.values()],
        **args
    ).interactive()
    return chart