为了账号安全,请及时绑定邮箱和手机立即绑定

带有twinx()的辅助轴:如何添加到图例?

/ 猿问

带有twinx()的辅助轴:如何添加到图例?

沧海一幻觉 2019-08-23 15:52:56

带有twinx()的辅助轴:如何添加到图例?

我有一个带有两个y轴的情节,使用twinx()。我也给线条贴了标签,并希望用它们来展示legend(),但我只是成功地在图例中获得了一个轴的标签:


import numpy as np

import matplotlib.pyplot as plt

from matplotlib import rc

rc('mathtext', default='regular')


fig = plt.figure()

ax = fig.add_subplot(111)

ax.plot(time, Swdown, '-', label = 'Swdown')

ax.plot(time, Rn, '-', label = 'Rn')

ax2 = ax.twinx()

ax2.plot(time, temp, '-r', label = 'temp')

ax.legend(loc=0)

ax.grid()

ax.set_xlabel("Time (h)")

ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")

ax2.set_ylabel(r"Temperature ($^\circ$C)")

ax2.set_ylim(0, 35)

ax.set_ylim(-20,100)

plt.show()

所以我只得到图例中第一个轴的标签,而不是第二个轴的标签“temp”。我怎么能将这第三个标签添加到图例中?

https://img1.mukewang.com/5d5f9b740001b61f08000562.jpg

查看完整描述

3 回答

?
互换的青春

您可以通过添加以下行来轻松添加第二个图例:

ax2.legend(loc=0)

你会得到这个:

https://img2.mukewang.com/5d5f9b87000168fb08150615.jpg

但是如果你想要一个传奇上的所有标签,那么你应该做这样的事情:


import numpy as np

import matplotlib.pyplot as plt

from matplotlib import rc

rc('mathtext', default='regular')


time = np.arange(10)

temp = np.random.random(10)*30

Swdown = np.random.random(10)*100-10

Rn = np.random.random(10)*100-10


fig = plt.figure()

ax = fig.add_subplot(111)


lns1 = ax.plot(time, Swdown, '-', label = 'Swdown')

lns2 = ax.plot(time, Rn, '-', label = 'Rn')

ax2 = ax.twinx()

lns3 = ax2.plot(time, temp, '-r', label = 'temp')


# added these three lines

lns = lns1+lns2+lns3

labs = [l.get_label() for l in lns]

ax.legend(lns, labs, loc=0)


ax.grid()

ax.set_xlabel("Time (h)")

ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")

ax2.set_ylabel(r"Temperature ($^\circ$C)")

ax2.set_ylim(0, 35)

ax.set_ylim(-20,100)

plt.show()

哪个会给你这个:

https://img1.mukewang.com/5d5f9b98000185ec08150615.jpg

查看完整回答
反对 回复 2019-08-23
?
慕雪6442864

从matplotlib 2.1版开始,您可以使用图例。可以创建图形图例ax.legend(),而不是使用轴的手柄ax生成图例


fig.legend(loc =“右上角”)

这将收集图中所有子图的所有句柄。由于它是一个图形图例,它将被放置在图的角落,并且loc参数是相对于图形的。


import numpy as np

import matplotlib.pyplot as plt


x = np.linspace(0,10)

y = np.linspace(0,10)

z = np.sin(x/3)**2*98


fig = plt.figure()

ax = fig.add_subplot(111)

ax.plot(x,y, '-', label = 'Quantity 1')


ax2 = ax.twinx()

ax2.plot(x,z, '-r', label = 'Quantity 2')

fig.legend(loc="upper right")


ax.set_xlabel("x [units]")

ax.set_ylabel(r"Quantity 1")

ax2.set_ylabel(r"Quantity 2")


plt.show()

https://img4.mukewang.com/5d5f9ba80001afcf04500336.jpg

为了将图例放回轴中,可以提供a bbox_to_anchor和a bbox_transform。后者将是图例应该存在的轴的轴变换。前者可以是由loc轴坐标给定的边的坐标。

fig.legend(loc="upper right", bbox_to_anchor=(1,1), bbox_transform=ax.transAxes)

https://img.mukewang.com/5d5f9bb40001727604500314.jpg

查看完整回答
反对 回复 2019-08-23
?
一只萌萌小番薯

您可以通过添加ax中的行来轻松获得所需内容:


ax.plot(0, 0, '-r', label = 'temp')

要么


ax.plot(np.nan, '-r', label = 'temp')

除了为斧头传说添加标签之外,这只会绘制一个标签。


我认为这是一种更容易的方法。当你在第二轴只有几条线时,没有必要自动跟踪线,因为像上面那样用手固定会很容易。无论如何,这取决于你需要什么。


整个代码如下:


import numpy as np

import matplotlib.pyplot as plt

from matplotlib import rc

rc('mathtext', default='regular')


time = np.arange(22.)

temp = 20*np.random.rand(22)

Swdown = 10*np.random.randn(22)+40

Rn = 40*np.random.rand(22)


fig = plt.figure()

ax = fig.add_subplot(111)

ax2 = ax.twinx()


#---------- look at below -----------


ax.plot(time, Swdown, '-', label = 'Swdown')

ax.plot(time, Rn, '-', label = 'Rn')


ax2.plot(time, temp, '-r')  # The true line in ax2

ax.plot(np.nan, '-r', label = 'temp')  # Make an agent in ax


ax.legend(loc=0)


#---------------done-----------------


ax.grid()

ax.set_xlabel("Time (h)")

ax.set_ylabel(r"Radiation ($MJ\,m^{-2}\,d^{-1}$)")

ax2.set_ylabel(r"Temperature ($^\circ$C)")

ax2.set_ylim(0, 35)

ax.set_ylim(-20,100)

plt.show()

情节如下:

https://img4.mukewang.com/5d5f9bc3000174c008000600.jpg

更新:添加更好的版本:

ax.plot(np.nan, '-r', label = 'temp')

plot(0, 0)可能会改变轴范围。


查看完整回答
反对 回复 2019-08-23

添加回答

回复

举报

0/150
提交
取消
意见反馈 邀请有奖 帮助中心 APP下载
官方微信