将2D矩阵从保存的.csv字符串转换为矩阵

Question

我一直在运行一些ML模型，并将结果(大小为[25，25，1](导出为.csv。当我稍后导入.csv时，列中的值是一个字符串，而不是值的矩阵。有人知道转换它的简单方法吗(最好不需要正则表达式通过括号或其他东西进行解析(？

Answer 1

您可以使用ast中的literal_eval将列表列表的字符串表示形式更改为实际列表，然后使用numpy 将其转换为数组

import ast
import numpy as np

a = "[[1,2], [3,4]]"
np.array(ast.literal_eval(a))
>>> array([[1, 2],
[3, 4]])