[python]读取txt数据并转为ndarray数组或list列表

自定义了一个函数，功能是读取纯数字文本的txt文件，并将数据存为数组（ndarray格式），函数可以直接调用使用。

import numpy as np

def txt2array(txt_path, delimiter):
    #---
    # 功能：读取只包含数字的txt文件，并转化为array形式
    # txt_path：txt的路径；delimiter：数据之间的分隔符
    #---
    data_list = []
    with open(txt_path) as f:
        data = f.readlines()
    for line in data:
        line = line.strip("\n")  # 去除末尾的换行符
        data_split = line.split(delimiter)
        temp = list(map(float, data_split))
        data_list.append(temp)

    data_array = np.array(data_list)
    return data_array

调用举例（delimiter指分隔符，假如txt中的数据用空格分割的）：

array = txt2array("yourfile.txt", " ")

若想将数据保存为list格式的，去掉 data_array = np.array(data_list) 这一行即可。