如何将单独的vlines添加到seaborn FacetGrid的每个子图中

import seaborn as sns
import matplotlib.pyplot as plt

# 加载数据集
data = sns.load_dataset('iris')

# 创建FacetGrid对象
g = sns.FacetGrid(data, col='species')

# 定义一个函数，用于在每个子图上添加垂直线
def add_vline(x, **kwargs):
    plt.vlines(x, 0, 1, **kwargs)  # x是垂直线的位置，0和1分别是y轴的起始和结束位置

# 使用map方法将add_vline函数应用到每个子图
g.map(add_vline, 'petal_length', color='red')  # 'petal_length'是要画线的位置的数据列，color是线条颜色

# 显示图形
plt.show()

def add_custom_vline(x, vline_positions, **kwargs):
    for pos in vline_positions:
        plt.vlines(pos, 0, 1, **kwargs)

# 假设我们有一个字典，包含了每个子图的垂直线位置
vline_positions = {'setosa': 4.5, 'versicolor': 5.0, 'virginica': 6.0}

# 修改FacetGrid的创建，以便传递额外的参数
g = sns.FacetGrid(data, col='species')
g.map(add_custom_vline, 'petal_length', vline_positions=vline_positions['species'], color='red')

plt.show()