python - 使用 python 从文本文件导入矩阵

Question

我有两个文本文件，其中写入了矩阵(不是 numpy 矩阵，所以它是列表的列表)。这些矩阵以字符串格式编写，因此文本文件如下所示:[[1,2,3],[3,4,5],[6,7,8]],[[3,3,3],[5,6,7],.....

我想使用 python 从文本文件中读回这个矩阵。我无法使用 numpy 进行读取，因为它给出 ValueError: Could not conversion string to float

有办法做到这一点吗？如果我首先将矩阵写为 numpy 矩阵会更容易吗(我需要为此更改以前程序的代码，并且只是想知道当矩阵存储为文本文件中的字符串)？

Answer 1

您可以使用 ast 模块:

import ast

strArray =  "[[1,2,3],[3,4,5],[6,7,8]]"

# evaluates the array in string format and converts it to a python array object
array = ast.literal_eval(strArray)

注意:对于像您这样的多个嵌套数组，literal_eval很可能会将字符串转换为以嵌套数组作为元素的元组。使用此模块时请记住这一点。