如何根据定义的组对树状图的标签进行着色？(在python中)

import matplotlib.pyplot as plt

# 定义树状图的数据
labels = ['A', 'B', 'C', 'D', 'E', 'F', 'G']
parents = [None, 'A', 'A', 'B', 'B', 'C', 'C']

# 定义每个标签对应的颜色
colors = {'A': 'red', 'B': 'blue', 'C': 'green', 'D': 'yellow', 'E': 'orange', 'F': 'purple', 'G': 'brown'}

# 创建树状图对象
fig, ax = plt.subplots()

# 绘制树状图
ax.bar(labels, [1]*len(labels), color=[colors[label] for label in labels])

# 设置图形属性
ax.set_xlabel('Labels')
ax.set_ylabel('Value')
ax.set_title('Colored Tree Diagram')

# 显示图形
plt.show()