import k3d
from math import sin, cos, pi
width = height = length = 20
model_matrix = [
7.0, 5.0, -5.0, 0.0,
0.0, 7.0, 7.0, 5.0,
7.0, -5.0, 5.0, 0.0,
0.0, 0.0, 0.0, 1.0
]
def f(x, y, z):
return sin(float(x) / width * pi * 2.0), cos(float(y) / height * pi * 2.0), sin(float(z) / length * pi * 2.0)
colors = (0xFF0000, 0x00FF00) * width * height * length
vectors = [[[f(x, y, z) for x in range(width)] for y in range(height)] for z in range(length)]
plot = k3d.plot()
plot += k3d.vector_field(vectors, colors, model_matrix=model_matrix, use_head=False)
plot.display()