plot_test.py

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
import math

from tb_points import points
g = (1 + 5**.5)/2
l = []
h = 1/g
u_rate = .1
v_rate = .3
w_rate = .2
x_orig, y_orig, z_orig = [], [], []
for p in points:
    x_orig.append(p[0])
    y_orig.append(p[1])
    z_orig.append(p[2])
x_orig = np.array(x_orig)
y_orig = np.array(y_orig)
z_orig = np.array(z_orig)
points = np.hstack((x_orig, y_orig, z_orig))
fig = plt.figure()
ax = plt.axes(projection='3d')

u = 0
v = 0
w = 0
x = [0]*20
y = [0]*20
z = [0]*20
plt.ion()
points = ax.scatter(x_orig, y_orig, z_orig)
#ax.set_zlim(-8192, 8192)
fig.canvas.draw_idle()
plt.pause(1)
##ax.set_xlim(-8192, 8192)
##ax.set_ylim(-8192, 8192)
##ax.set_zlim(-8192, 8192)
##plt.pause(1)
##u = math.asin(0.0625)
##print(2048*math.cos(u), 2048*math.sin(u))
##v = math.asin(-0.0625)
##w = math.asin(0.0625)
##for i in range (0, 20):
##    x[i] = (math.cos(u) * math.cos(v))*x_orig[i] + (math.cos(u) * math.sin(v)*math.sin(w) - math.sin(u)*math.cos(w))*y_orig[i] + (math.cos(u)*math.sin(v)*math.cos(w)+math.sin(u)*math.sin(w))*z_orig[i]
##    y[i] = (math.sin(u) * math.cos(v))*x_orig[i] + (math.sin(u) * math.sin(v)*math.sin(w) + math.cos(u)*math.cos(w))*y_orig[i] + (math.sin(u)*math.sin(v)*math.cos(w)-math.cos(u)*math.sin(w))*z_orig[i]
##    z[i] = -math.sin(v)*x_orig[i] + math.cos(v)*math.sin(w)*y_orig[i] + math.cos(v)*math.cos(w)*z_orig[i]
####points = ax.scatter(x,y,z)
####fig.canvas.draw_idle()
##plt.pause(1)
##
##points = ax.scatter(r_x, r_y, rz)


for t in range(30):
##        for i in range (0, 20):
##            x[i] = (math.cos(u) * math.cos(v))*x_orig[i] + (math.cos(u) * math.sin(v)*math.sin(w) - math.sin(u)*math.cos(w))*y_orig[i] + (math.cos(u)*math.sin(v)*math.cos(w)+math.sin(u)*math.sin(w))*z_orig[i]
##            y[i] = (math.sin(u) * math.cos(v))*x_orig[i] + (math.sin(u) * math.sin(v)*math.sin(w) + math.cos(u)*math.cos(w))*y_orig[i] + (math.sin(u)*math.sin(v)*math.cos(w)-math.cos(u)*math.sin(w))*z_orig[i]
##            z[i] = -math.sin(v)*x_orig[i] + math.cos(v)*math.sin(w)*y_orig[i] + math.cos(v)*math.cos(w)*z_orig[i]
        x = [series[t][i][0] for i in range(20)]
        y = [series[t][i][1] for i in range(20)]
        points.remove()
        points = ax.scatter(x,y)
        fig.canvas.draw_idle()
        plt.pause(.01)