python - 将列表元素声明为符号 sympy python

Question

我有一个函数可以在某些条件下返回一些列表元素：

 m1=   [[1, 0], [0, 1]]
 m2=   [[2, 0], [0, 2]]

我想做的是将 m1 和 m2 声明为包含这些索引的符号，即 m1[0] 是符号 [1,0]，使用 Sympy，但到目前为止我的尝试没有成功。

有谁知道我该怎么做？

任何帮助将不胜感激！！

Answer 1

我猜你想使用symarray。

from sympy import symarray

m1 = symarray('', 2)
m1[0] = [1, 0]