関数でnumpy、sympyを使用している象徴マトリックス？

Question

マイコード：

def crearMatrix(name,shape=(2,2)): 
    X = np.empty(shape) 
    for i in range(shape[0]): #X.shape[1] 
     for j in range(shape[1]): 
      X[i][j] = Symbol("a"+'_{'+str(i*10+j+11)+'}') 
    return X 

エラーメッセージ：

TypeError: can't convert expression to float

Answer 1

おそらくnumpy matrix'esがsympyのシンボルを保存する必要はありません。このためにはsympy.Matrixを使用してください。

import sympy 

def crearMatrix(name,shape=(2,2)): 
    X = [] 
    for i in range(shape[0]): 
     row = [] 
     for j in range(shape[1]): 
      row.append(sympy.Symbol("a"+'_{'+str(i*10+j+11)+'}')) 
     X.append(row) 
    return sympy.Matrix(X) 

A = crearMatrix("a") 
print(sympy.latex(A**2)) 

# \left[\begin{matrix}a_{11}^{2} + a_{12} a_{21} & a_{11} a_{12} + a_{12} a_{22}\\a_{11} a_{21} + a_{21} a_{22} & a_{12} a_{21} + a_{22}^{2}\end{matrix}\right]