如何从文件中的单列或带索引数据生成指定维度矩阵、四阶张量

场景1：单列下三角浮点数转对称矩阵 #

问题说明 #

存储的是n阶对称矩阵按行优先展开的下三角元素（含对角线），示例中3个元素对应2阶矩阵。

错误原因 #

原有代码循环内错误覆盖循环变量i、j，每次循环读取两行，完全不符合下三角元素的读取逻辑。

正确实现代码 #

import numpy as np

# 读取单列所有浮点值
with open("final_overlap.txt", "r") as f:
    vals = [float(line.strip()) for line in f if line.strip()]
# 由元素总数反推矩阵阶数n：n(n+1)/2 = 元素总数
n = int((np.sqrt(8 * len(vals) + 1) - 1) / 2)
mat = np.zeros((n, n))
# 填充下三角区域
ptr = 0
for i in range(n):
    for j in range(i + 1):
        mat[i][j] = vals[ptr]
        ptr += 1
# 补全上三角，减去重复计算的对角线值
mat = mat + mat.T - np.diag(mat.diagonal())
print(mat)

场景2：带行列索引的下三角数据转对称矩阵 #

问题说明 #

每行格式为行索引 列索引 值，仅存储下三角区域数据，示例中10个元素对应4阶对称矩阵。

正确实现代码 #

import numpy as np

with open("indexed_lower_tri.txt", "r") as f:
    lines = [line.strip().split() for line in f if line.strip()]
# 取最大行索引确定矩阵阶数
max_dim = max(int(line[0]) for line in lines)
mat = np.zeros((max_dim, max_dim))
# 逐行赋值到对应位置
for line in lines:
    i = int(line[0]) - 1
    j = int(line[1]) - 1
    val = float(line[2])
    mat[i][j] = val
# 补全上三角
mat = mat + mat.T - np.diag(mat.diagonal())
print(mat)

场景3：多索引数据转高维对称张量 #

问题说明 #

每行格式为索引1 索引2 索引3 索引4 值，通常满足置换对称性，示例中6个元素对应2×2×2×2四阶对称张量。

错误原因 #

原有代码用二维列表存储四维数据，填充逻辑完全不匹配高维对称规则，导致数组越界。

正确实现代码 #

import numpy as np

with open("Two_Electron.txt", "r") as f:
    lines = [line.strip().split() for line in f if line.strip()]
# 自动计算各维度大小（取对应索引的最大值）
dim_count = len(lines[0]) - 1
dims = [max(int(line[k]) for line in lines) for k in range(dim_count)]
tensor = np.zeros(dims)
# 逐行赋值并按对称规则补全所有等价位置
for line in lines:
    idxs = [int(line[k]) - 1 for k in range(dim_count)]
    val = float(line[-1])
    i,j,k,l = idxs
    # 按四阶电子积分常用的8种对称置换规则补全，可根据实际对称规则修改
    tensor[i,j,k,l] = val
    tensor[j,i,k,l] = val
    tensor[i,j,l,k] = val
    tensor[j,i,l,k] = val
    tensor[k,l,i,j] = val
    tensor[l,k,i,j] = val
    tensor[k,l,j,i] = val
    tensor[l,k,j,i] = val
print(tensor)

通用实现思路（支持任意维度） #

1. 读取所有有效数据行，拆分得到索引和对应数值
2. 统计每个维度的最大索引值，确定张量的维度大小
3. 初始化对应维度的全零张量
4. 逐行给索引对应位置赋值，再根据数据的对称规则补全所有对称位置的数值
5. 输出最终张量即可

内容的提问来源于stack exchange，提问作者Abd-Elazeem Mohamed