PANDAS -如何根据索引中的值对整行进行着色

import pandas as pd

data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}

index_labels = ['ID1', 'ID2', 'ID3']

df = pd.DataFrame(data, index=index_labels)

def color_rows(val):
    color = 'background-color: yellow' if val.name == 'ID2' else ''
    return [color] * len(val)

styled_df = df.style.apply(color_rows, axis=1)

styled_df