Matplotlib Python 3 не рисует стрелки, начинающиеся с не-источника - PullRequest
0 голосов
/ 06 октября 2019

У меня есть следующий код, который строит векторы слов:

import numpy as np
import matplotlib.pyplot as plt
la = np.linalg

words = ['I', 'like', 'enjoy', 'deep', 'learning', 'NLP', 'flying', '.']
X = np.array([ [0,2,1,0,0,0,0,0],
               [2,0,0,1,0,1,0,0],
               [1,0,0,0,0,0,1,0],
               [0,1,0,0,1,0,0,0],
               [0,0,0,1,0,0,0,1],
               [0,1,0,0,0,0,0,1],
               [0,0,1,0,0,0,0,1],
               [0,0,0,0,1,1,1,0]])
U, s, Vh = la.svd(X, full_matrices = False)
ax = plt.axes()
for i in range(len(words)): 
    plt.text(U[i,0],U[i,1],words[i])
    ax.arrow(0,0,U[i,0],U[i,1],head_width=0.1, head_length=0.1, fc='lightblue', ec='black')
plt.xlim(-.8,.2)
plt.ylim(-.8,.8)
plt.grid()

plt.title(' Simple SVD word vectors in Python',fontsize=10)
plt.show()
plt.close()

enter image description here

Это рисует стрелки, начиная с начала координат. Тем не менее, когда я пытаюсь построить если из других точек.

ax.arrow(-0.8,-0.8,U[i,0],U[i,1],head_width=0.1, head_length=0.1, fc='lightblue', ec='black')

enter image description here

Стрелки не рисуются! В чем проблема, пожалуйста?

Спасибо.

1 Ответ

1 голос
/ 06 октября 2019

Использование matplotlib.axes.Axes.arrow:

  • Это рисует стрелку от (x, y) до (x + dx, y + dy)
  • Настройка(-0,8, -0,8), получает следующее

enter image description here

  • Чтобы изменить направление стрелки, необходимочтобы компенсировать ненулевое происхождение
  • со следующей строкой:
ax.arrow(-0.8, -0.8, (U[i,0] + 0.8), (U[i,1] + 0.8),head_width=0.1, head_length=0.1, fc='lightblue', ec='black')
  • Получит вам это:

enter image description here

...