Python_squarify_Legend 基于块的颜色

如何解决Python_squarify_Legend 基于块的颜色

我正在根据以下 df 创建树状图，其中每个 country 在 color 列中都有自己的颜色：

import pandas as pd  
import matplotlib
import matplotlib.pyplot as plt
import squarify 
  
data = {'name': ['marcus Raboy','Jay Karas','Cathy Garcia-Molina',\
                 'Jay Chapman','Raúl Campos,Jan Suter','David Dhawan',\
                 'Shannon Hartman','Ryan Polito','Hakan Algül','Martin Scorsese'],\
        'counrty':['United States','United States','Philippines',\
                   'Mexico','India','Turkey','United States'],\
        'color': ['yellow','yellow','blue','green','red','grey','yellow'],'show_cnt': [15,14,13,12,9,8,8]}  

df = pd.DataFrame(data)  

我在这个片段中绘制了树状图：

fig = plt.gcf()
ax = fig.add_subplot()
fig.set_size_inches(23,10)
squarify.plot(sizes=df.show_cnt,label=df.director,color=df.color,alpha=.5)
plt.axis('off')
plt.show()

我尝试将此代码用于图例：

plt.legend(handles=df.color,labels=df.country)

输出：

ValueError: The truth value of a Series is ambiguous. Use a.empty,a.bool(),a.item(),a.any() or a.all().

如何根据颜色-国家模式渲染图例？