自定义车削标记的实时绘图

Real-time plotting of a custom turning marker

是否可以以某种方式交互式地绘制自定义标记(如 this),但它是否实时转动?散点图似乎没有授予对标记的任何访问权限。

您可以使用 FancyArrowPatch 创建自定义标记。许多样式和选项都是可能的。这样的补丁不容易更新,但是你可以删除补丁并重新创建它来创建动画。

创建动画最简单的方法是通过 plt.pause(),但这并不适用于所有环境。另一种方法是通过 FuncAnimation,它涉及更多行,但可以更轻松地控制动画。

下面是一些示例代码来展示这些概念:

import matplotlib.pyplot as plt
from matplotlib import patches
from matplotlib.collections import PatchCollection
from matplotlib import animation
import numpy as np

fig, ax = plt.subplots()

N = 50
x = np.random.uniform(-20, 20, (N, 2))
dx = np.random.uniform(-1, 1, (N, 2))
dx /= np.linalg.norm(dx, axis=1, keepdims=True)
colors = plt.cm.magma(np.random.uniform(0, 1, N))

arrow_style = "Simple,head_length=2,head_width=3,tail_width=1"

ax.set_xlim(-40, 40)
ax.set_ylim(-30, 30)
ax.set_aspect('equal')

old_arrows = None

def animate(i):
    global old_arrows, x, dx

    if old_arrows is not None:
        old_arrows.remove()
    x += dx
    dx += np.random.uniform(-.1, .1, (N, 2))
    dx /= np.linalg.norm(dx, axis=1, keepdims=True)
    arrows = [patches.FancyArrowPatch((xi, yi), (xi + dxi * 10, yi + dyi * 10), arrowstyle=arrow_style)
              for (xi, yi), (dxi, dyi) in zip(x, dx)]
    old_arrows = ax.add_collection(PatchCollection(arrows, facecolors=colors))
    return old_arrows,

ani = animation.FuncAnimation(fig, animate, np.arange(1, 200),
                              interval=25, repeat=False, blit=True)
plt.show()

我通过 remove() 和这样的静态变量解决了它:

class pltMarker:
    def __init__(self, angle=None, pathString=None):
        self.angle = angle or []
        self.pathString = pathString or """simply make and svg, open in a text editor and copy the path XML string in here"""
        self.path = parse_path( self.pathString )
        self.path.vertices -= self.path.vertices.mean( axis=0 )
        self.marker = mpl.markers.MarkerStyle( marker=self.path )
        self.marker._transform = self.marker.get_transform().rotate_deg(angle)

    def rotate(self, angle=0):
        self.marker._transform = self.marker.get_transform().rotate_deg(angle)

def animate(k):

    angle = ... # new angle
    myPltMarker.rotate(angle)

    animate.Scatter.remove()
    animate.Scatter = plt.scatter(1, 0, marker=myPltMarker.marker, s=100)

    return animate.Scatter, 

angle = ...
myPltMarker = pltMarker(angle=angle)
animatePlt.Scatter = plt.scatter(1, 0, marker=myPltMarker.marker, s=100)   

anm = animation.FuncAnimation(fig, animate, blit=False, interval=1)
plt.show()