Question

我有一个无法解决的问题。我尝试了几种方法，但是没有用。或至少我很愚蠢，以获得解决方案idk。

用for循环创建子图，每次用子数组中的数据创建子图。我为每个散点创建了不同的颜色（每个子图大约13个散点）。

现在，我想为每个子图创建一个图例，以显示在哪个点上实现哪种颜色。

fig, axs = plt.subplots(3, 3, figsize=(15, 15), facecolor='w', edgecolor='k')
fig.subplots_adjust(hspace = .35, wspace= .3)

C = np.array([[255, 0, 0], [0, 255, 0], [0, 0, 255], [0, 255, 255], [255, 255, 0], [0, 128, 0], [139, 0, 139], [255, 140, 0], [216, 112, 147], [238, 130, 238], [119, 136, 128], [0, 0, 0], [139, 69, 19]])
axs = axs.ravel()

for i in range(9):



    SingleWeldLayer = np.where(WeldArray1[:,0] == PlotLayerStart + i)
    Last = np.amax(SingleWeldLayer)
    First = np.amin(SingleWeldLayer)

    DynArray = WeldArray1[First:Last+1]

#    print(DynArray)

    axs[i].scatter(DynArray[:,3], DynArray[:,2], c = C/255)

    axs[i].set_title('Layer: '+ str(PlotLayerStart + i))
    axs[i].set_xlabel('MF_LargeCoil_TargetPosition', fontsize=8)
    axs[i].set_ylabel('HF_SmallCoil_TargetPosition', fontsize=8)


plt.show()

有人可以帮我为每个子图创建图例吗？

Current scatter plots

Answer 1

在循环中，添加axs[i].legend(loc="upper right")

子图散点图的解决方案

1 个答案: