VPythonでピラミッド

Pythonの3Dグラフィックスライブラリ、VPythonの簡単すぎる習作として球をピラミッド状に配置にするものを作ってみた。ソースはこんなん。

# -*- coding: cp932 -*-
from visual import color
from visual import vector
from visual import sphere

unit = 2
limit = 20
for i in range(limit + 1):
    for x in range(i): 
      for y in range(i):
        tpos = vector((x - 0.5*i) * unit ,(y - 0.5*i)*unit,-i*unit)
        tcol = color.green
        if (i == limit and (x == 0 or y == 0 or x == (i-1) or y == (i-1))) or \
           ((x == 0 or x == (i - 1)) and (y == 0 or y == (i - 1))):
            tcol = color.blue
        sphere(color=tcol,pos=tpos)

みじか。簡単なことしかしてないけど、簡単なことを簡単に書けるのは大事。結果は以下のスクリーンショット。ピラミッドの上のほうから見下ろしているかんじ。球の色は基本は緑色で、ピラミッドのふちだけ青色に。マウス操作でぐりぐりと回転拡大縮小もできる。