リスト内包表記を使い numpy の 一次元配列 (ndarray) を作成します．
import numpy as np arr = np.array([x**2 for x in range(5)]) print(arr)
[ 0 1 4 9 16]