如何在Numpy矩阵中使用字母作为变量构造符号矩阵？

报错原因 #

Numpy 是纯数值计算库，矩阵元素必须是已赋值的具体数值对象，不能直接使用未定义的变量作为占位符，所以你直接写未赋值的 x、y、z 会触发未定义报错。

实现任意实数占位符的方法 #

你需要做群性质验证这类符号运算，应该搭配 SymPy 符号计算库实现，操作步骤如下：

1. 导入库并定义实值符号变量

import numpy as np
import sympy as sp

# 定义x、y、z为实数类型的符号变量
x, y, z = sp.symbols('x y z', real=True)

2. 构造符号形式的3×3上三角矩阵

# 直接用SymPy的Matrix构造符号矩阵，完全支持矩阵运算
G_a = sp.Matrix([[1, x, z], [0, 1, y], [0, 0, 1]])

数值运算转换方法 #

如果你需要代入具体实数做数值验证，可以用subs方法赋值后转成numpy数组使用：

# 代入x=1, y=2, z=3的具体值，转numpy浮点数组
G_a_np = np.array(G_a.subs({x:1, y:2, z:3}), dtype=np.float64)

提示：np.matrix已经被Numpy官方标记为弃用，新代码建议用np.array处理数值矩阵，符号运算直接用SymPy的Matrix即可。

群性质验证示例 #

你描述的这类矩阵属于海森堡群，本身是非阿贝尔群，你可以通过以下代码验证交换律不成立：

# 定义第二组参数的符号矩阵
a, b, c = sp.symbols('a b c', real=True)
G_b = sp.Matrix([[1, a, c], [0, 1, b], [0, 0, 1]])

# 计算两种顺序的矩阵乘积
prod_ab = G_a * G_b
prod_ba = G_b * G_a

# 化简后对比两个乘积的差，差不为零即证明不满足交换律
print(sp.simplify(prod_ab - prod_ba))

内容的提问来源于stack exchange，提问作者Veneratu