リスト内包表記を使い numpy の 三次元配列 (ndarray) を作成します．
import numpy as np i = range(100, 400, 100) j = range(10, 40, 10) k = range(1, 4, 1) arr = np.array([[[ix+jy+kz for kz in k] for jy in j] for ix in i]) print(arr)
[ 0 1 4 9 16]