Из вашего вопроса, я не на 100% ясно, что вы хотите делать.
Вы хотите просто пометить каждую вершину в строке? Или вы хотите только отметить вершины, которые являются целыми числами? Или вы хотите интерполировать, где целые «пересечения» будут пересекаться вдоль линии и маркировать их?
Прежде всего, для загрузки текстового файла, просмотрите numpy.loadtxt
, если вы еще этого не сделали. В вашем конкретном случае вы можете сделать что-то вроде:
z, x, y = np.loadtxt('data.txt', usecols=[0, 5, 6]).T
Во всяком случае, как быстрый пример простейшего варианта (маркировка каждой вершины):
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(10)
y = 2 * x
z = x ** 2
fig, ax = plt.subplots()
ax.plot(x, y, 'bo-')
for X, Y, Z in zip(x, y, z):
# Annotate the points 5 _points_ above and to the left of the vertex
ax.annotate('{}'.format(Z), xy=(X,Y), xytext=(-5, 5), ha='right',
textcoords='offset points')
plt.show()
[/g0]
Теперь, для второго варианта, у нас может быть что-то более похожее (похоже на то, что предлагалось @ Mathematical.coffee):
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(-0.6, 5.6, 0.2)
y = 2 * x
z = x**2
fig, ax = plt.subplots()
ax.plot(x, y, 'bo-')
# Note the threshold... I'm assuming you want 1.000001 to be considered an int.
# Otherwise, you'd use "z % 1 == 0", but beware exact float comparisons!!
integers = z % 1 < 1e-6
for (X, Y, Z) in zip(x[integers], y[integers], z[integers]):
ax.annotate('{:.0f}'.format(Z), xy=(X,Y), xytext=(-10, 10), ha='right',
textcoords='offset points',
arrowprops=dict(arrowstyle='->', shrinkA=0))
plt.show()
[/g1]