如何在 matplotlib treemap 中获取形状坐标？

Question

我有可用的树状图，我需要获取此树状图的每个形状的坐标，例如之后将它们放入 GeoJSON 中。有没有什么功能可以帮助我，或者我将从这个树状图的 svg 版本中获取所有坐标？

Answer 1

与ax = squarify.plot(...)、ax.patches包含Rectangle patches的列表。那些 Rectangle 具有 get_x() 等功能。坐标在轴坐标系中，在 x 和 y 方向上似乎都从 0 到 100。

当您在同一图中绘制更多元素时，ax 可能还包含其他元素，因此您可能需要过滤它们。

import matplotlib.pyplot as plt
import squarify  # pip install squarify (algorithm for treemap)

ax = squarify.plot(sizes=[13, 22, 35, 5], label=["group A", "group B", "group C", "group D"], color=['b','r','y','g'])

for rect in ax.patches:
    x, y, w, h = rect.get_x(), rect.get_y(), rect.get_width(), rect.get_height()
    c = rect.get_facecolor()
    print(f'Rectangle x={rect.get_x()} y={rect.get_y()} w={rect.get_width()} h={rect.get_height()} ')
plt.axis('off')
plt.show()

PS：获取相应的文本，再次假设绘图只包含树图：

for rect, text in zip(ax.patches, ax.texts):
    x, y, w, h = rect.get_x(), rect.get_y(), rect.get_width(), rect.get_height()
    c = rect.get_facecolor()
    t = text.get_text()

如何在 matplotlib treemap 中获取形状坐标？

How to get coordinates of shape in matplotlib treemap?

python

matplotlib

treemap

squarify