Python读取TXT文件计算数组均值报错及解决需求

DevOps 小满

2026-7-16

问题描述

代码：

import ast
import numpy as np

file_path = "C:/Users/USER/OneDrive - Technion/Research_Technion/Python_PNM/Basics/Velocity_Average/Test.txt"
with open(file_path) as f:
    content = f.read()
    data = ast.literal_eval(content)

for arr in data:
    arr = np.array(arr)  # Convert the array to a NumPy array
    filtered_data = arr[arr > 1e-9]  # Filter elements greater than 1e-9
    mean = np.mean(filtered_data)
    print(mean)

Test.txt内容：

[array([[1.0],
       [2.0],
       [3.0],
       [4.0],
       [1e-12]])]
[array([[1.0],
       [2.0],
       [3.0],
       [4.0],
       [1e-12]])]

报错信息：

line 7
[array([[1.0],
^
SyntaxError: invalid syntax

期望输出：[2.5, 2.5]

解决方案

问题根源有两点：

Test.txt中的array(...)是NumPy专属语法，并非Python原生字面量，ast.literal_eval无法解析该格式。
文件内的两个列表彼此独立，未被包裹成一个完整的可解析Python数据结构（如大列表），直接解析会触发语法错误。

修正步骤

修改Test.txt内容：将array(...)替换为普通嵌套列表，并把两个子列表放入一个大列表中，修正后内容如下：

[
    [[1.0], [2.0], [3.0], [4.0], [1e-12]],
    [[1.0], [2.0], [3.0], [4.0], [1e-12]]
]

优化代码输出格式：原代码逻辑可正常运行，若要直接输出期望的列表格式，可调整代码如下：

import ast
import numpy as np

file_path = "C:/Users/USER/OneDrive - Technion/Research_Technion/Python_PNM/Basics/Velocity_Average/Test.txt"
with open(file_path) as f:
    content = f.read()
    data = ast.literal_eval(content)

means = []
for arr in data:
    arr = np.array(arr)
    filtered_data = arr[arr > 1e-9]
    mean = np.mean(filtered_data)
    means.append(mean)

print(means)