FancyArrowPatch類將路徑作為參數,所以我認為你可以使用它.

1)對于每個線段,創建一個matplotlib.path.Path實例.

2)使用路徑實例繪制箭頭.

import numpy as np

import matplotlib.pyplot as plt; plt.ion()

from matplotlib.patches import FancyArrowPatch, PathPatch

from matplotlib.path import Path

def create_path(x,y):

vertices = zip(x,y)

codes = [Path.MOVETO] + (len(vertices)-1) * [Path.CURVE3]

return Path(vertices, codes)

X = np.linspace(0,4*np.pi,10000)

Y = np.sin(X)

fig, ax = plt.subplots(1,1)

ax.plot(X,Y,color='blue')

shift = 0.1

seg_size = 300

i = 0

while i +seg_size < len(X):

x = X[i:i+seg_size]

y = Y[i:i+seg_size]+shift

path = create_path(x,y)

# for testing path

# patch = PathPatch(path, facecolor='none', lw=2)

# ax.add_patch(patch)

arrow = FancyArrowPatch(path=path, color='r')

ax.add_artist(arrow)

i += seg_size*2

不幸的是,這不起作用,因為傳遞給FancyArrowPatch的路徑不能超過2個段(沒有記錄,但是在ensure_quadratic_bezier中有一個檢查).

所以你必須作弊.下面我使用每個段的最后2個點來繪制箭頭.

import numpy as np

import matplotlib.pyplot as plt; plt.ion()

from matplotlib.patches import FancyArrowPatch

X = np.linspace(0,4*np.pi,10000)

Y = np.sin(X)

fig, ax = plt.subplots(1,1)

ax.plot(X,Y,color='blue')

shift = 0.1

seg_size = 300

i = 0

while i +seg_size < len(X):

x = X[i:i+seg_size]

y = Y[i:i+seg_size]+shift

ax.plot(x, y, 'k')

posA, posB = zip(x[-2:], y[-2:])

edge_width = 2.

arrowstyle = "fancy,head_length={},head_width={},tail_width={}".format(2*edge_width, 3*edge_width, edge_width)

arrow = FancyArrowPatch(posA=posA, posB=posB, arrowstyle=arrowstyle, color='k')

ax.add_artist(arrow)

i += seg_size*2

總結

以上是生活随笔為你收集整理的python单行箭头_python – 如何在matplotlib的曲线末端放置一个箭头？的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。