matplotlib - pyplot：更改图例中的 ncols

Question

我正在绘制一个大型数据集和 pyplot 中的一些回归。数据根据附加值着色。我决定将图例中的列数设置为 2。

数据点看起来不错，但对于回归，我想回到 ncols=1。有可能在一个传说中做到这一点吗？

（我知道，我可以宣布两个传说，但我想避免这种情况......）

Answer 1

matplotlib 中的图例是 OffsetBox 的集合。原则上，您可以通过手动重新排列这些偏移框来更改图例中的“ncol”。当然，这需要一些 matplotlib 内部知识，对于普通用户来说会比较困难。不知道什么是暴露这一点的最佳方式。这是一个简单的代码，它创建两个图例并将它们组合成一个。

import matplotlib.pyplot as plt
import matplotlib.legend as mlegend

ax = plt.subplot(111)
l1, = ax.plot([1,2,3])

leg1 = ax.legend([l1], ["long lable"], ncol=1)

leg2 = mlegend.Legend(ax, [l1, l1, l1, l1], tuple("1234"), ncol=2)

leg1._legend_box._children.append(leg2._legend_box._children[1])
leg1._legend_box.align="left" # the default layout is 'center'

plt.show()