如何遍历某一列以创建多个图python

import pandas as pd
import matplotlib.pyplot as plt

data = pd.read_csv('data.csv')  # 假设数据保存在data.csv文件中

column_name = 'column_name'  # 替换为你要遍历的列名
unique_values = data[column_name].unique()  # 获取列中的唯一值

for value in unique_values:
    subset = data[data[column_name] == value]  # 根据列值筛选子集
    # 创建图表
    plt.plot(subset['x'], subset['y'], label=value)  # 假设x和y是数据中的两列
    plt.xlabel('x')
    plt.ylabel('y')
    plt.title('Graph for ' + value)
    plt.legend()

plt.show()  # 显示图表