python – 如何将现有Line2D对象的样式复制到plot()调用？ (matplotlib)

2019年7月21日 552次阅读

我需要提取matplotlib.lines.Line2D对象的样式信息,以便在matplotlib.pyplot.plot()调用中使用它.并且(如果可能的话)我希望以比Line2D.properties()输出中过滤与样式相关的属性更优雅的方式来实现它.

代码可能是这样的：

import matplotlib.pyplot as plt

def someFunction(a, b, c, d, **kwargs):
    line = plt.plot(a, b, marker='x', **kwargs)[0]
    plt.plot(c, d, marker='o', **kwargs) # the line I need to change

在这种情况下,我想要使用相同的样式(包括颜色)绘制两条线,但使用不同的标记.
此外,我希望能够使用plot()函数的’autocolouring’功能,除非颜色已明确作​​为关键字参数给出.

最佳答案 plt.plot()返回的Line2D对象具有update_from()方法,该方法将原始实例中的所有属性复制到新实例,但仅保留该行的数据.您可以使用此行复制所有属性,然后设置应该“手动”不同的所有属性.这里有一个小例子：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D


def someFunction(a, b, c, d, *args, **kwargs):
    line1, = plt.plot(a, b, marker='x', *args, **kwargs)
    line2, = plt.plot(c, d) # the line I need to change

    ##copy properties from line1
    line2.update_from(line1)

    ##set every that should differ *after* update_from()
    line2.set_marker('o')


x1 = np.linspace(0,np.pi,10)
x2 = np.linspace(np.pi,2*np.pi,10) 
y1 = -np.sin(x1)**2
y2 = np.sin(x2)**2

someFunction(x1,y1,x2,y2, '--', lw=3, ms=10, color='g')

plt.show()

这给出了以下图片：

《python – 如何将现有Line2D对象的样式复制到plot()调用？ (matplotlib)》

如果省略color关键字,将使用autocolouring.

点赞