如何使用 sympy 图的标记参数？

Question

sympy plot 命令有一个 markers 参数：

markers ：指定所需标记类型的字典列表。字典中的键应该相当于 matplotlib 的plot()函数的参数以及与标记相关的关键字参数。

如何使用markers参数？我失败的尝试范围从

from sympy import *
x = symbols ('x')
plot (sin (x), markers = 'o')

到

plot (sin (x), markers = list (dict (marker = 'o')))

Answer 1

不错的发现！

文档并没有把事情说清楚。深入研究源代码，会导致 图中的这些行.py：

            for marker in parent.markers:
                # make a copy of the marker dictionary
                # so that it doesn't get altered
                m = marker.copy()
                args = m.pop('args')
                ax.plot(*args, **m)

所以，sympy 只是调用 matplotlib 的 plot 其中：

• args 键字典作为位置参数
• 字典的所有其他键作为关键字参数

由于 matplotlib 的 plot 允许使用各种各样的参数，因此这里都支持它们。它们的主要目的是在绘图上显示额外的标记（您需要给出它们的位置）。

示例：

from sympy import symbols, sin, plot

x = symbols('x')
plot(sin(x), markers=[{'args': [2, 0, 'go']},
                      {'args': [[1, 3], [1, 1], 'r*'], 'ms': 20},
                      {'args': [[2, 4, 6], [-1, 0, -1], ], 'color': 'turquoise', 'ls': '--', 'lw': 3}])

这些内容会转换为：

ax.plot(2, 0, 'go')  # draw a green dot at position 2,0
ax.plot([3, 5], [1, 1], 'r*', ms=20)  # draw red stars of size 20 at positions 3,1 and 5,1
ax.plot([2, 4, 6], [-1, 0, -1], ], color='turquoise', ls='--', lw=3)
    # draw a dotted line from 2,-1  over 4,0 to 6,-1

PS：源代码显示了带有注释、矩形和填充的字典的类似方法（使用plt.fill Between()):

        if parent.annotations:
            for a in parent.annotations:
                ax.annotate(**a)
        if parent.rectangles:
            for r in parent.rectangles:
                rect = self.matplotlib.patches.Rectangle(**r)
                ax.add_patch(rect)
        if parent.fill:
            ax.fill_between(**parent.fill)

Answer 2

（更新使用@cards建议（在评论中）和这篇文章。

您可以使用 Matplotlib 后端将标记直接添加到轴。

import sympy as sp
from sympy.plotting.plot import MatplotlibBackend

# plot with sympy
x = sp.symbols('x')

p0 = sp.plot(sp.sin(x),(x,-sp.pi,sp.pi),line_color='b',legend=True,show=False)

# point to be displayed
x0 = float(sp.pi/2)
y0 = float(sp.sin(sp.pi/2))

# plot with the commands from the backend (default matplotlib)
be = MatplotlibBackend(p0)
be.process_series()

plt = be.plt
plt.plot([x0,-x0],[y0,-y0],'r*',markersize=10,label="Star marker")

# Update the legend to include markers
plt.legend()

plt.show()

此解决方案还使用标记更新图例。