在numpy中表示粒子运动

Numpy很好地处理了这个案子,使用 structured arrays . 对于您的特定情况,您可以按如下方式创建一个数组(假设您希望矩阵为3x3以便与SVD一起使用)。 编辑 :应该是 np.zeros 若要创建数组,则可以逐行或逐字段复制数据。

d = dtype([('pos', float, 3), ('v', float, 3), ('Matrix', float, (3, 3)), ('det', float)])
arr = np.zeros(1000, dtype=d).view(np.recarray)

recarray ,但它允许将结构化数组字段作为属性访问(即。, arr.Matrix

arr.det = [det(a) for a in arr.Matrix] 

您还可以使用括号表示法和字段名称访问字段,这适用于重新排列或结构化数组。

arr['det'] = [det(a) for a in arr['Matrix']]