本文介绍如何通过继承 `np.ndarray` 创建专用于对称矩阵的自定义数组类，自动强制对称性、保持赋值时的对称约束，并合理利用 `numpy.linalg.eigh` 进行特征分解，避免冗余存储。
在科学计算中，对称矩阵（如协方差矩阵、哈密顿量、图拉普拉斯等）具有重要性质：实对称矩阵必可正交对角化，且其特征值为实数、特征向量正交。为在 NumPy 生态中安全、高效地利用这些性质，推荐通过子类化 np.ndarray 构建专用类型——SymmetricArray。
该类需满足两个核心要求：
- 构造时自动对称化：将任意输入矩阵 $ M $ 映射为对称部分 $ \frac{1}{2}(M + M^\top) $；
- 赋值时保持对称性：对任意位置或切片的写入操作，同步更新其转置位置，确保结构不变。
以下是最简可行实现（兼容 2D 及高维批量矩阵，如 (N, D, D)）：
import numpy as np
class SymmetricArray(np.ndarray):
def __new__(cls, input_array):
input_array = np.asarray(input_array)
assert input_array.ndim >= 2 and input_array.shape[-1] == input_array.shape[-2], \
"Last two dimensions must be square"
# 对最后两维取对称部分；其余维度广播处理
axes = list(range(input_array.ndim - 2)) + [-1, -2]
transposed = input_array.transpose(axes)
sym_part = 0.5 * (input_array + transposed)
return sym_part.view(cls)
def __setitem__(self, key, value):
# 标准化索引为 tuple，补全至 ndim
if not isinstance(key, tuple):
key = (key,)
if len(key) < self.ndim:
key += (slice(None),) * (self.ndim - len(key))
# 构造转置索引：仅交换最后两维的下标
key_t = key[:-2] + (key[-1], key[-2])
# 确保 value 与 key 兼容；若 value 是矩阵，也需对称赋值
value = np.asarray(value)
super().__setitem__(key, value)
# 若 value 非标量且最后两维存在，对其转置后赋给对称位置
if value.ndim >= 2 and len(key) >= 2:
value_t = value.transpose(*list(range(value.ndim - 2)) + [-1, -2])
super().__setitem__(key_t, value_t)
elif value.ndim == 0 or len(key) < 2:
super().__setitem__(key_t, value)✅ 使用示例：
rng = np.random.default_rng(42)
A = SymmetricArray(rng.random((3, 3)))
print("初始对称矩阵：\n", A)
A[0, :] = [1, 2, 3] # 自动同步更新第0列 → A[:, 0] = [1,2,3]
print("赋值后：\n", A)
# 输出保证 A[0,1]==A[1,0], A[0,2]==A[2,0], etc.⚠️ 重要注意事项：
- 不建议将 U, D 作为实例属性缓存：特征分解结果依赖于当前数组值，而 NumPy 子类无法自动追踪底层数据变更（如 += 或切片赋值）。强行缓存易导致状态不一致。
- ✅ 正确做法是按需调用 np.linalg.eigh：它专为实对称/复厄米特矩阵优化，返回正交特征向量和实特征值，且数值稳定：
U, D = np.linalg.eigh(A) # D 是一维数组，U 是正交矩阵 # 验证重建：np.allclose(A, U @ np.diag(D) @ U.T)
- 避免重载 __array_finalize__ 或 __array_wrap__：本实现仅需 __new__ 和 __setitem__ 即可满足核心需求；过度定制易引入不可预见的广播或视图行为。
总结：SymmetricArray 不是“魔法容器”，而是语义增强的契约型数组——它不改变 NumPy 的底层机制，而是通过构造与赋值阶段的显式约束，保障用户始终在对称流形上操作。配合 eigh、cholesky 等专用函数，可构建更健壮、可读性更强的数值代码。
