pandas – テーブルの表示をカスタマイズする方法

import numpy as np
import pandas as pd
from IPython.display import display, HTML

df = pd.DataFrame(
    [
        [11, None, 15, 4, 5],
        [2, 12, 10, 8, 16],
        [17, 24, 14, 13, 23],
        [19, 6, 18, None, 7],
        [3, 9, 20, 21, 1],
    ],
    index=["a", "b", "c", "d", "e"],
    columns=["A", "B", "C", "D", "E"],
)
df

print(type(df.style))

style = df.style.highlight_null()
print(type(style))

style

style = (
    df.style.highlight_null().highlight_min(color="green").highlight_max(color="blue")
)
style

style = df.style.highlight_null("red")

df.style.highlight_min(color="green")

df.style.highlight_max(color="green")

df = pd.DataFrame(
    {
        "A": [132, -107, -162, 96, 145, -133, 12, 35, 168, -12],
        "B": [0, -143, 21, 10, 105, 56, 120, 103, -132, 63],
        "C": [-31, 56, 67, 0, 16, 139, 0, -38, 142, -58]
    }
)

df.style.bar(color=["red", "green"], align="mid")

df = pd.DataFrame(
    np.arange(25).reshape(5, 5),
    index=["a", "b", "c", "d", "e"],
    columns=["A", "B", "C", "D", "E"],
)
# 列ごと
display(df.style.background_gradient(axis=0))

# 行ごと
display(df.style.background_gradient(axis=1))

df = pd.DataFrame(
    [
        [11, None, 15, 4, 5],
        [2, 12, 10, 8, 16],
        [17, 24, 14, 13, 23],
        [19, 6, 18, None, 7],
        [3, 9, 20, 21, 1],
    ],
    index=["a", "b", "c", "d", "e"],
    columns=["A", "B", "C", "D", "E"],
)

df.style.set_precision(2)

df2 = (df / df.sum(axis=0))

df2.style.format("{:.0%}", na_rep="-")

df2.style.format({"A": "{:.0%}", "B": "{:.2%}"}, na_rep="-")

df.style.set_na_rep("-")

df.style.set_caption("タイトル")

df = pd.DataFrame([[1, 2], [3, 4]], index=["a", "b"], columns=["A", "B"])

style = df.style.background_gradient()
HTML(style.render())

style = df.style.background_gradient().set_uuid("hoge")
HTML(style.render())

style = df.style.background_gradient()
display(style)

style.clear()
display(style)

df = pd.DataFrame(
    np.arange(25).reshape(5, 5),
    index=["a", "b", "c", "d", "e"],
    columns=["A", "B", "C", "D", "E"],
)

# 特定の列
display(df.style.background_gradient(axis=0, subset=["A", "C"]))

# 特定の行
display(df.style.background_gradient(axis=1, subset=pd.IndexSlice[["a", "c"], :]))

# 特定の範囲
display(df.style.background_gradient(axis=1, subset=pd.IndexSlice["a":"c", "A":"C"]))

style = df.style.set_table_styles(
    [{"selector": "tr:hover", "props": [("background-color", "yellow")]}]
)
style

style = df.style.set_properties(
    **{"background-color": "black", "color": "lawngreen", "border-color": "white"}
)

style

def highlight_max(s):
    return ["background-color: green" if x == s.max() else "" for x in s]


df.style.apply(highlight_max)

def highlight_even(x):
    return "background-color: green" if x % 2 == 0 else ""


df.style.applymap(highlight_even)

df.style.where(lambda x: x > 10, "background-color: green", "background-color: red")

def set_min_max(style):
    return style.highlight_min(color="red").highlight_max(color="red")


def set_format(style):
    return style.format("{:.2f}")


df.style.pipe(set_min_max).pipe(set_format)