如何解决使用 squareify.plot 在标签上显示多列值
我有一个数据框,我想用 squarify
绘制树状图。我想通过编辑 country_name
参数在图表上显示 counts
和 labels
,但它似乎只取一个值。
示例数据
import squarify
import pandas as pd
from matplotlib import pyplot as plt
d = {'country_name':['USA','UK','Germany'],'counts':[100,200,300]}
dd = pd.DataFrame(data=d)
fig = plt.gcf()
ax = fig.add_subplot()
fig.set_size_inches(16,4.5)
norm = matplotlib.colors.normalize(vmin=min(dd.counts),vmax=max(dd.counts))
colors = [matplotlib.cm.Blues(norm(value)) for value in dd.counts]
squarify.plot(label=dd.country_name,sizes=dd.counts,alpha=.7,color=colors)
plt.axis('off')
plt.show()
预期输出将在图表上同时包含 counts
和 country_name
。
解决方法
您可以通过同时循环遍历两列并组合组合字符串来创建标签列表。例如:
import squarify
import pandas as pd
from matplotlib import pyplot as plt
import matplotlib
d = {'country_name': ['USA','UK','Germany'],'counts': [100,200,300]}
dd = pd.DataFrame(data=d)
labels = [f'{country}\n{count}' for country,count in zip(dd.country_name,dd.counts)]
fig = plt.gcf()
ax = fig.add_subplot()
fig.set_size_inches(16,4.5)
norm = matplotlib.colors.Normalize(vmin=min(dd.counts),vmax=max(dd.counts))
colors = [matplotlib.cm.Blues(norm(value)) for value in dd.counts]
squarify.plot(label=labels,sizes=dd.counts,alpha=.7,color=colors)
plt.axis('off')
plt.show()
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。