Anaconda Enterprise: Documentation

# categorical_heatmap_periodic.py

    from bokeh.io import output_file, show
    from bokeh.models import ColumnDataSource, HoverTool
    from bokeh.plotting import figure
    from bokeh.sampledata.periodic_table import elements
    from bokeh.transform import dodge, factor_cmap

    output_file("periodic.html")

    periods = ["I", "II", "III", "IV", "V", "VI", "VII"]
    groups = [str(x) for x in range(1, 19)]

    df = elements.copy()
    df["atomic mass"] = df["atomic mass"].astype(str)
    df["group"] = df["group"].astype(str)
    df["period"] = [periods[x-1] for x in df.period]
    df = df[df.group != "-"]
    df = df[df.symbol != "Lr"]
    df = df[df.symbol != "Lu"]

    cmap = {
        "alkali metal"         : "#a6cee3",
        "alkaline earth metal" : "#1f78b4",
        "metal"                : "#d93b43",
        "halogen"              : "#999d9a",
        "metalloid"            : "#e08d49",
        "noble gas"            : "#eaeaea",
        "nonmetal"             : "#f1d4Af",
        "transition metal"     : "#599d7A",
    }

    source = ColumnDataSource(df)

    p = figure(plot_width=900, plot_height=500, title="Periodic Table (omitting LA and AC Series)",
               x_range=groups, y_range=list(reversed(periods)), toolbar_location=None, tools="")

    p.rect("group", "period", 0.95, 0.95, source=source, fill_alpha=0.6, legend="metal",
           color=factor_cmap('metal', palette=list(cmap.values()), factors=list(cmap.keys())))

    text_props = {"source": source, "text_align": "left", "text_baseline": "middle"}

    x = dodge("group", -0.4, range=p.x_range)

    r = p.text(x=x, y="period", text="symbol", **text_props)
    r.glyph.text_font_style="bold"

    r = p.text(x=x, y=dodge("period", 0.3, range=p.y_range), text="atomic number", **text_props)
    r.glyph.text_font_size="8pt"

    r = p.text(x=x, y=dodge("period", -0.35, range=p.y_range), text="name", **text_props)
    r.glyph.text_font_size="5pt"

    r = p.text(x=x, y=dodge("period", -0.2, range=p.y_range), text="atomic mass", **text_props)
    r.glyph.text_font_size="5pt"

    p.text(x=["3", "3"], y=["VI", "VII"], text=["LA", "AC"], text_align="center", text_baseline="middle")

    p.add_tools(HoverTool(tooltips = [
        ("Name", "@name"),
        ("Atomic number", "@{atomic number}"),
        ("Atomic mass", "@{atomic mass}"),
        ("Type", "@metal"),
        ("CPK color", "$color[hex, swatch]:CPK"),
        ("Electronic configuration", "@{electronic configuration}"),
    ]))

    p.outline_line_color = None
    p.grid.grid_line_color = None
    p.axis.axis_line_color = None
    p.axis.major_tick_line_color = None
    p.axis.major_label_standoff = 0
    p.legend.orientation = "horizontal"
    p.legend.location ="top_center"

    show(p)