python - 从Python文件中读取数组

Question

我有一个文件，其中写入了一个数组，例如:

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

如何使用 python 读取它并获取变量中的数组？到目前为止，我只得到了字符串。

def get_data():
     myfile = open('random_array.txt', 'r')
     data = myfile.read().replace('\n', '')
     return data

Answer 1

如果格式总是这样，一种方法是使用 json.loads:

>>> s = "[1,2,3,4]"
>>> import json
>>> json.loads(s)
[1, 2, 3, 4]

这样做的优点是您可以在逗号之间使用任何空格，可以在文本中使用 float 和整数，等等。

因此，根据您的情况，您可以这样做:

import json

def get_data():
    with open("random_array.txt", "r") as f:
        return json.load(f)