Python+matplotlib自定义坐标轴位置、颜色、箭头

时间:2024-04-08 13:58:24

容易的事。例如下面的代码:

Python+matplotlib自定义坐标轴位置、颜色、箭头

 

绘制结果为:

Python+matplotlib自定义坐标轴位置、颜色、箭头

虽然确实几行代码就画出了正弦曲线,但是这个图也太朴素了,如果我们想得到下面这样的结果,该如何实现呢?

Python+matplotlib自定义坐标轴位置、颜色、箭头

 

 

上面图形中的难点在于坐标轴的箭头,可以使用axisartist来辅助实现,参考代码:

Python+matplotlib自定义坐标轴位置、颜色、箭头

 

虽然这个axisartist能够很方便地自定义坐标轴的位置和完美地设置箭头,但对于更多属性的设置有点乏力,很多参数的微调还是很花费时间的,在上面代码中添加代码修改坐标轴颜色,效果并不是很理想。例如,

Python+matplotlib自定义坐标轴位置、颜色、箭头

 

绘制结果如下:

Python+matplotlib自定义坐标轴位置、颜色、箭头

 

也可以考虑隐藏原有的坐标轴直线,使用arrow来绘制带箭头的直线来代替,就是箭头有点丑,调整参数也比较麻烦,例如下面的代码:

Python+matplotlib自定义坐标轴位置、颜色、箭头

结果如图所示:

Python+matplotlib自定义坐标轴位置、颜色、箭头

 

如果实现下面的图形,上面几种方法还是有些吃力的:

Python+matplotlib自定义坐标轴位置、颜色、箭头

分析图形中的细节可知,主要是坐标轴的属性,有:1)只显示两根轴线;2)两根轴线颜色不同;3)轴线宽度比默认值粗;4)坐标轴的刻度颜色与默认值不同;5)轴线的一端有表示方向的箭头。

这些属性都可以通过轴域的坐标轴方法来修改,再借助于annotate来实现箭头,很容易就做出效果来了。参考代码如下:

Python+matplotlib自定义坐标轴位置、颜色、箭头