如何为cpairs图中的颜色添加图例

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd

# 假设我们有一个DataFrame df，其中包含我们要绘制的数据
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}
df = pd.DataFrame(data)

# 使用seaborn的cpairs图功能
g = sns.PairGrid(df)
g.map_diag(sns.histplot)
g.map_offdiag(sns.scatterplot)

# 添加颜色图例
# 假设我们根据某个条件给数据点着色，这里简单使用'A'列的值
colors = df['A'].apply(lambda x: 'red' if x < 2 else 'blue')

# 创建一个颜色条
handles = [plt.Line2D([0], [0], marker='o', color='w', markerfacecolor=color, markersize=10) for color in set(colors)]
labels = list(set(colors))
plt.legend(handles, labels, title="Color Legend")

plt.show()