Маркировка расстояния между точками на графике питона - PullRequest
Новая
       64

Маркировка расстояния между точками на графике питона

0 голосов
/ 21 октября 2018

У меня есть список точек, перечисленных в pts.along с линиями, соединяющими эти точки, чтобы сделать тетраэдр.Я пытаюсь найти способ добавить метки, показывающие евклидово расстояние между каждой точкой. 

fig = plt.figure()
ax = fig.gca(projection='3d')

# Plot points
pts = [(a,0,0),(b,0,0),(c1,c2,0),(z1,z2,z3)]
for p in pts:
    ax.scatter(p[0], p[1], p[2], zdir='z', c='r')

# Plot tetrahedron
for a, b in itertools.product(pts, pts):
    x = np.linspace(a[0], b[0], 100)
    y = np.linspace(a[1], b[1], 100)
    z = np.linspace(a[2], b[2], 100)
    ax.plot(x, y, z)

Любая помощь приветствуется.

1 Ответ

0 голосов
/ 21 октября 2018

Вы можете рассчитать расстояния и затем использовать text для их отображения.

Здесь они приземляются прямо на линии, но с трехмерным графиком и проекциями было бы трудно автоматически поместить их в лучшее место.Также обратите внимание, что использование product создает дубликаты точек, поэтому я показываю расстояние только с d>0.

enter image description here

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import itertools
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')

# Plot points
a, b, c1, c2, z1, z2, z3 = 1, 2, 3, 4, 3, 6, 3
pts = [(a,0,0),(b,0,0),(c1,c2,0),(z1,z2,z3)]
for p in pts:
    ax.scatter(p[0], p[1], p[2], zdir='z', c='r')

# Plot tetrahedron
for a, b in itertools.product(pts, pts):
    x = np.linspace(a[0], b[0], 100)
    y = np.linspace(a[1], b[1], 100)
    z = np.linspace(a[2], b[2], 100)
    d = np.sqrt( sum([(a[i]-b[i])**2 for i in (0, 1, 2)]) )
    s = "%.2f" % d
    m = [(a[i]+b[i])/2. for i in (0, 1, 2)]
    ax.plot(x, y, z)
    if d>0:
        ax.text(m[0], m[1], m[2], s)
    print a, b, s

plt.show()
...