从文件中读取二维数组

Question

我正在使用 Python 进行模拟项目，但遇到了问题。我需要将二维数组保存到文件中，以便稍后使用其中的数据。初始列表如下所示：

int_array3 = [[14,5], [9,3], [25,5]]
#writing to file:
with open('data.txt', 'w') as f:
for num in int_array3:
    f.write("%s\n" % num)

文件中的数据如下所示：

[14  5]
[9 3]
[25  6]

这段代码看腻了，但它不是整数列表。

with open('data.txt', 'r') as f:
result = f.readlines()
result = [x.strip() for x in result]

结果列表如下所示：

['[14  5]', '[9 3]', '[25  6]']

有人可以推荐我如何让它工作吗？我希望结果列表的形状与我的初始列表相同。我正在使用 numpy。

Answer 1

您可以尝试使用 eval() 将字符串转换为数组，如下所示：

A = ['[14  5]', '[9 3]', '[25  6]']
arr = []

for i in range(len(A)):
    l = eval(A[i].split()[0] + ',' + A[i].split()[1])
    arr.append(l)

print(A)
# [[14, 5], [9, 3], [25, 6]]

请随时要求澄清

Answer 2

仅将没有[] 的数字写入文件怎么样？您可以在写入文件期间使用嵌套的for 循环来实现这一点。只要记住分开数字（例如用空格）。 然后，您可以逐行读取文件，使用split()，然后您将获得一个数字数组，但现在它是一个字符串数组。要从中生成数字，您需要将字符串数字转换为 int。我认为最简单的方法是使用列表推导来构建新的数字列表。

为了说明上述情况，您编写的代码将是这样的：

int_array3 = [[14,5], [9,3], [25,5]]
#writing to file:
with open('data.txt', 'w') as f:
for vec in int_array3:
    for number in vec:
        f.write("%d " % number)
    f.write("\n")

你可以读懂你的诡计：

with open('data.txt', 'r') as f:
for line in f:
    vector = [int(x) for x in line.split()]
    #Do something with vector variable