如何保持 label 只出现一次但在其中使用迭代数字

Question

我正在尝试创建一个 animation，它将显示一条基于当前半径向上移动图形的线。 现在，我的代码正在为每个图形创建一个新的 label，我无法弄清楚如何使用 label 内部的迭代变量 i 让 label 读取“当前半径 = i”，其中 i 是迭代的索引，而不是字母 i，如果这有意义的话。 我已经在下面发布了我的代码及其 output，非常感谢您的帮助！

list = np.linspace(0,1,12)  # vector of increasing radii
death_list = np.array( (.1, .2, .3, .4, .8) )   # list of radii that cause deaths

fig, (ax1, ax2) = plt.subplots(1, 2)
camera = Camera(fig)

for i in list :
    ax1.scatter(x,y, color = "green", marker = "o", s = 15)

    for xx,yy in zip(x,y) :
        cir = mpatches.Circle((xx, yy), i, color='r',fill=True, zorder = 0)
        ax1.add_patch(cir)

    ax1.set_aspect('equal', adjustable='datalim')
    ax1.set_title('Random Points on a Circle')
    ax1.set(xlabel='x-axis', ylabel='y-axis')
    ax1.set_xlim([-1.25, 1.25])
    ax1.set_ylim([-1.25, 1.25])

    a = np.arange(0,1.1,.1)
    b = a
    ax2.plot(a,b, color = 'b')
    ax2.axhline(y = i, label = f'current radius = {i}', color = 'r', linestyle = '-')

    if i in death_list :
        ax2.plot(0, i, marker="o", markersize = 5, markeredgecolor = "green", markerfacecolor = "green")
    
    ax2.set_aspect('equal', adjustable='datalim')
    ax2.set_title('Persistence Diagram')
    ax2.set(xlabel='Birth', ylabel='Death')
    ax2.legend(loc = 'lower right')

Answer 1

如果我正确理解您的问题，您希望通过以下替换在图例中标记每一行： ax2.axhline(y = i, label = 'current radius = ', color = 'r', linestyle = '-') → ax2.axhline(y = i, label = 'current radius = {0}'.format(i), color = 'r', linestyle = '-')

如果这不是您要找的，请告诉我。 我最近处理了类似的问题。

编辑：@matszwecja 我不知道你可以直接这样做。 你每天学习新的东西。

二次编辑：@Piethon，在下面的评论中回答您的问题Also, is there a way to truncate the decimal?

是的，用我的方法，很简单。 如果您想要N个小数位等，只需将{0}替换为{0:.Nf} 。例如{0:.2f}将 output 保留两位小数。

至于你的其他问题Ok, awesome that helped, I have updated the question with the new code and the new output, it still outputting the label multiple times. should I place the ax2?legend(loc = 'lower right') line outside of the for loop? Ok, awesome that helped, I have updated the question with the new code and the new output, it still outputting the label multiple times. should I place the ax2?legend(loc = 'lower right') line outside of the for loop? ，我不太确定你的意思。 尽管在您的代码中确实有很多不必要的重复。 例如：

ax1.set_aspect('equal', adjustable='datalim')
ax1.set_title('Random Points on a Circle')
ax1.set(xlabel='x-axis', ylabel='y-axis')
ax1.set_xlim([-1.25, 1.25])
ax1.set_ylim([-1.25, 1.25])
...

ax2.set_aspect('equal', adjustable='datalim')
ax2.set_title('Persistence Diagram')
ax2.set(xlabel='Birth', ylabel='Death')
ax2.legend(loc = 'lower right')

都可以放在循环之外。

Answer 2

我会使用 f-string 格式：

label = f'current radius = {i}'