排除由for loop with matplotlib创建的图例中的地块

import matplotlib.pyplot as plt

# 创建一个空的图形对象和子图
fig, ax = plt.subplots()

# 创建一些示例数据和标签
data = [1, 2, 3, 4]
labels = ['A', 'B', 'C', 'D']

# 用for循环创建图块
for i in range(len(data)):
    ax.plot(data[i], label=labels[i])

# 获取当前图形对象的图例
handles, labels = ax.get_legend_handles_labels()

# 定义一个处理函数，用于排除特定的图块
def exclude_patch(label):
    if label == 'A':  # 排除标签为'A'的图块
        return False
    return True

# 使用处理函数来过滤图例中的图块
filtered_handles = [handle for i, handle in enumerate(handles) if exclude_patch(labels[i])]

# 创建新的图例
ax.legend(filtered_handles, labels, loc='upper right')

# 显示图形
plt.show()