Python 读取TXT文本数据并存为 array

方法一 str -> list -> array

import numpy as np
f = open(r"D:\data.txt")
line = f.readline()
data_list = []
while line:
    num = list(map(float,line.split()))
    data_list.append(num)
    line = f.readline()
f.close()
data_array = np.array(data_list)

上边的程序主要是将txt文本中的数据首先通过文件读出转换为list，最后再转换为array

函数	功能
readline()	用于从文件读取整行
split()	通过指定分隔符对字符串进行切片
map()	map() 会根据提供的函数对指定序列做映射。第一个参数 function 以参数序列中的每一个元素调用 function 函数，返回包含每次 function 函数返回值的新列表

方法二 str -> array

import numpy as np
f = open(r"D:\data.txt")
line = f.readline()
data = np.zeros((100,4))
i = 0
while line:
    num = np.array([float(x) for x in line.split()])
    data[i,:] = num
    line = f.readline()
    i = i+1
f.close()

这种方法可以直接将文件中的数据转换为array，但是需要提前知道文件中数组的维度，不如第一种方法简便。