在下面的python代码中,我正在绘制数据帧中的时间数据和多个y值。现在我想: - 垂直旋转x轴的时间值 - 将所有y标签(y1-y4)移动到轴的顶部是否有人有建议或解决方案?
import pandas as pd
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
import matplotlib.pyplot as plt
data = {'time':['00:00:00','18:00:00','23:59:00'],
'y1': [1,2,3],'y2': [4,5,6],'y3': [7,8,9],'y4': [10,11,12]}
df=pd.DataFrame(data,columns=['time','y1','y2','y3','y4'])
df['time']=pd.to_datetime(df['time'],format='%H:%M:%S')
host=host_subplot(111,axes_class=AA.Axes)
plt.subplots_adjust(right=0.75)
par1=host.twinx()
par2=host.twinx()
par3=host.twinx()
offset1=40
offset2=80
new_fixed_axis=par2.get_grid_helper().new_fixed_axis
par2.axis['right']=new_fixed_axis(loc='right',axes=par2,offset=(offset1,0))
par2.axis['right'].toggle(all=True)
par3.axis['right']=new_fixed_axis(loc='right',axes=par3,offset=(offset2,0))
par3.axis['right'].toggle(all=True)
host.set_xlabel('Time')
host.set_ylabel('y1')
par1.set_ylabel('y2')
par2.set_ylabel('y3')
par3.set_ylabel('y4')
p1,=host.plot(df['time'],df['y1'])
p2,=par1.plot(df['time'],df['y2'])
p3,=par2.plot(df['time'],df['y3'])
p4,=par3.plot(df['time'],df['y4'])
host.set_ylim(0,5)
par1.set_ylim(0,8)
par2.set_ylim(0,10)
par3.set_ylim(0,15)
host.legend(loc='upper left',bbox_to_anchor=(0,-.15),ncol=4)
host.axis['left'].label.set_color(p1.get_color())
host.axis["left"].label.set_rotation(90)
par1.axis['right'].label.set_color(p2.get_color())
par1.axis["right"].label.set_rotation(-90)
par2.axis['right'].label.set_color(p3.get_color())
par2.axis["right"].label.set_rotation(-90)
par3.axis['right'].label.set_color(p4.get_color())
par3.axis["right"].label.set_rotation(-90)
plt.draw()
plt.show()
我有第一个问题的解决方案,第二个问题是一个丑陋的黑客。
1)旋转x轴文本
你需要使用host.axis
属性设置它,然后向下移动图例和xlabel以腾出空间:
host.legend(loc='upper left',bbox_to_anchor=(0,-.3),ncol=4) # Lower legend a bit
host.axis["bottom"].label.set_pad(50) # Lower x-label a bit
host.axis["bottom"].major_ticklabels.set( # This is the actual rotation command
rotation=90, verticalalignment='center', horizontalalignment='right', pad=-2)
2)将y标签移动到轴的顶部
我在这里尝试的没有任何工作。具体来说,我希望以下工作:
par1.axis["right"].label.set_position((1.0, 1.0))
但是相应的get_position
命令保持返回(1.0, 0.5)
,其中0.5是轴的中间。由于某种原因,'set'命令根本不会坚持。这是丑陋的黑客入侵的地方 - 在标签下方添加空白行。因此,在设置标签文本的位置,请更改为:
new_line_pos_fix = ''.join(['\n'] * 9) # 9 is the number of blank lines; change as needed
host.set_ylabel('y1' + new_line_pos_fix)
par1.set_ylabel('y2' + new_line_pos_fix)
par2.set_ylabel('y3' + new_line_pos_fix)
par3.set_ylabel('y4' + new_line_pos_fix)
现在我们只需要设置标签的垂直对齐方式,使空行生效,添加一些填充以提高可读性:
host.axis['right'].label.set(verticalalignment='bottom', pad=7)
par1.axis['right'].label.set(verticalalignment='bottom', pad=7)
par2.axis['right'].label.set(verticalalignment='bottom', pad=7)
par3.axis['right'].label.set(verticalalignment='bottom', pad=7)
如果你想要更有组织的代码,上面的命令可以与颜色和旋转设置结合使用,例如:
par1.axis['right'].label.set(color=p2.get_color(), rotation=-90, verticalalignment='bottom', pad=7)
结果: