#!/usr/bin/env python
# coding: utf-8

# In[ ]:


from k3d import K3D
from math import sin, cos, pi

width = height = length = 20
view_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 z in range(length)] for y in range(height)] for x in range(width)]

plot = K3D()
plot += K3D.vectors_fields(vectors, colors, view_matrix=view_matrix, use_head=False)
plot.display()