from k3d import K3D
from math import sqrt, sin, cos
width = height = length = 100
def r(x, y, z):
r = sqrt((x - width / 2) * (x - width / 2) + (y - height / 2) * (y - height / 2) + (z - length / 2) * (z - length / 2))
r += sin(x / 2) * 3
r += cos(y / 10) * 5
return r
def f(x, y, z):
return 0 if r(x, y, z) > width / 2 else 1 if y + sin(x / 20) * 10 > height / 2 else 2
color_map = (0xffff00, 0xff0000)
voxels = [[[f(x, y, z) for z in range(length)] for y in range(height)] for x in range(width)]
plot = K3D()
plot += K3D.voxels(voxels, color_map)
plot.display()