python - 将二维数组读入Python的简单方法

Question

我的一个程序的输出是一个二维数组，例如

[[1,2,3],[4,5,6,7],[8,9]]

长度不确定。我将此输出作为单行写入文件 geno_matrix，然后需要将其读入另一个程序进行分析和处理。我需要将其作为另一个二维数组读入另一个程序。

显然，按 , 分割是行不通的，因为我最终会得到一个括号和单个数字的列表。如果我这样做

for line in geno_matrix:
    vec = line.rstrip('\n').split(',')

这将分隔所有条目(和括号)。我也尝试过

vec = line.rstrip('\n').split(',[')

返回一个列表

['[[1,2,3],[4,5,6,7],[8,9]]']

这样 vec[0] 是字符串(不是数组)'[[1,2,3],[4,5,6,7],[8, 9]]'，这让我回到了最初的分割问题。使用 Python 读取原始二维列表的最简单方法是什么？

Answer 1

您可以使用ast的literal_eval函数:

from ast import literal_eval
literal_eval('[[1,2,3], [4,5,6,7], [8,9]]')

会给你[[1,2,3], [4,5,6,7], [8,9]] 作为数组。