python - 将格式化字符串转换为二维数组

Question

这个问题已经有答案了:

How to convert string representation of list to a list (21 个回答)

已关闭 2 年前。

我有一个字符串，其格式类似于二维数组，如下所示:

"[[1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0]]"

如何让程序将字符串读取为实际的二维数组，如下所示

[[1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0]]

这样我就可以进行矩阵计算？

我尝试删除逗号和括号，然后将字符串数字转换为整数，但这不起作用，因为我需要保持二维数组的形状。在此示例中，我需要最终的 2D 数组大小为 [4][4]。

Answer 1

鉴于 2D 列表字符串恰好是 Python 2D 列表的有效语法，您可以只使用 eval() 函数:

output = eval("[[1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0]]")
print(output)
print(output[0])

打印:

[[1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0]]
[1, 0, 0, 0]