更改xytext以防止批注重叠

2024-05-18 21:41:45 发布

您现在位置:Python中文网/ 问答频道 /正文

所以我有一些代码可以生成一个甜甜圈图,但问题是有些情况下注释会因值而重叠。下面是代码和问题。在

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

def donut_chart(val):

    df_vals = pd.DataFrame.from_dict(val, orient='index')
    labels = df_vals.index.tolist()

    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(aspect="equal"))

    color = ['grey']*20
    color[0] = 'red'

    wedges, texts, junk = ax.pie(df_vals[0:4], counterclock = True, 
                            wedgeprops=dict(width=0.6, linewidth = 2, edgecolor = 'w'), 
                           startangle=90, colors=color,
                           autopct='%1.0f%%',
                           pctdistance=0.75,
                           textprops={'fontsize': 14})

    bbox_props = dict(boxstyle="square,pad=0.3", fc="w", ec="w", lw=0.72)
    kw = dict(xycoords='data', textcoords='data', arrowprops=dict(arrowstyle="-"),
              bbox=bbox_props, zorder=0, va="center")

    for i, p in enumerate(wedges):
        ang = (p.theta2 - p.theta1)/2. + p.theta1
        y = np.sin(np.deg2rad(ang))
        x = np.cos(np.deg2rad(ang))
        horizontalalignment = {-1: "right", 1: "left"}[int(np.sign(x))]
        connectionstyle = "angle,angleA=0,angleB={}".format(int(ang))
        kw["arrowprops"].update({"connectionstyle": connectionstyle})
        ax.annotate(labels[i], xy=(x, y), xytext=(1.2*np.sign(x), 1.2*y),
                     horizontalalignment=horizontalalignment, **kw, size=14)

    #centre_circle = plt.Circle((0,0),0.5, fc='white',linewidth=1.25)
    #fig.gca().add_artist(centre_circle)
    plt.axis('equal')
    plt.show()
    plt.close()


val = {'Label A':50, 'Label B':2, 'Label C':1, 'Label D':0.5}
donut_chart(val)

问题:

enter image description here

我想做的是创造这样的东西:

enter image description here

关键点似乎是改变xytext中的y值,这样标签就不会重叠,但我一直在想这是如何实现的,甚至是它是否可行。在

有什么想法吗?在


Tags: importdfasnppltvalaxlabel
1条回答
网友
1楼 · 发布于 2024-05-18 21:41:45

更新代码

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

def donut_chart(val):

    df_vals = pd.DataFrame.from_dict(val, orient='index')
    labels = df_vals.index.tolist()

    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(aspect="equal"))

    color = ['grey']*20
    color[0] = 'red'

    wedges, texts = ax.pie(df_vals[0:5], counterclock = True, 
                            wedgeprops=dict(width=0.6, linewidth = 1, edgecolor = 'w'), 
                           startangle=90, colors=color,
                           textprops={'fontsize': 14})

    bbox_props = dict(boxstyle="square,pad=0", fc="w", ec="w", lw=0.72)
    kw = dict(xycoords='data', textcoords='data', arrowprops=dict(arrowstyle="-"),
              bbox=bbox_props, zorder=0, va="center")

    for i, p in enumerate(wedges):
        ang = (p.theta2 - p.theta1)/2. + p.theta1
        y = np.sin(np.deg2rad(ang))
        x = np.cos(np.deg2rad(ang))
        horizontalalignment = {-1: "right", 1: "left"}[int(np.sign(x))]
        connectionstyle = "angle,angleA=0,angleB={}".format(int(ang))
        kw["arrowprops"].update({"connectionstyle": connectionstyle})
        ax.annotate(labels[i], xy=(x, y), xytext=((i/10) + 1.1*np.sign(x), (i/10) + y),
                     horizontalalignment=horizontalalignment, **kw, size=14)

    #centre_circle = plt.Circle((0,0),0.5, fc='white',linewidth=1.25)
    #fig.gca().add_artist(centre_circle)
    plt.axis('equal')
    plt.show()
    plt.close()
val = {'Label A':50, 'Label B':2, 'Label C':0.2, 'Label D':0.2,'Label E':0.2}
donut_chart(val)

差异

^{pr2}$

output

相关问题 更多 >

    热门问题