如何使用Python将文本文件读入列表或数组
我想读取一个文本文件的行到python列表或数组。 我只需要能够在创build后单独访问列表或数组中的任何项目。
文本文件格式如下:
0,0,200,0,53,1,0,255,...,0.
如果...
在上面,则实际的文本文件具有数百或数千个项目。
我正在使用下面的代码来尝试将文件读入列表中:
text_file = open("filename.dat", "r") lines = text_file.readlines() print lines print len(lines) text_file.close()
我得到的输出是:
['0,0,200,0,53,1,0,255,...,0.'] 1
显然,它将整个文件读入一个项目的列表,而不是单个项目的列表。 我究竟做错了什么?
你将不得不使用split()
将你的string分割成一个值列表
所以,
lines = text_file.read().split(',')
python的file.readLines()方法返回文件中行的列表:
f = open('file_name.ext', 'r') x = f.readlines()
现在你应该可以迭代x行的数组了。
你也可以使用numpy的loadtxt
from numpy import loadtxt lines = loadtxt("filename.dat", comments="#", delimiter=",", unpack=False)
所以你想创build一个列表清单…我们需要从一个空的列表开始
list_of_lists = []
接下来,我们逐行阅读文件内容
with open('data') as f: for line in f: inner_list = [elt.strip() for elt in line.split(',')] # in alternative, if you need to use the file content as numbers # inner_list = [int(elt.strip()) for elt in line.split(',')] list_of_lists.append(inner_list)
一个常见的用例是柱状数据,但是我们的存储单元是文件的行,我们已经逐一读取,因此您可能需要转置列表的列表。 这可以用下面的习语来完成
by_cols = zip(*list_of_lists)
另一个常用的用法是给每一列命名
col_names = ('apples sold', 'pears sold', 'apples revenue', 'pears revenue') by_names = {} for i, col_name in enumerate(col_names): by_names[col_name] = by_cols[i]
以便您可以在同类数据项上进行操作
mean_apple_prices = [money/fruits for money, fruits in zip(by_names['apples revenue'], by_names['apples_sold'])]
我写的大部分内容可以使用标准库中的csv
模块进行加速。 另一个第三方模块是pandas
,它可以让你自动化一个典型的数据分析的大多数方面(但有一些依赖)。
更新在Python 2中, zip(*list_of_lists)
返回一个不同的(转置的)列表列表,在Python 3中情况已经改变, zip(*list_of_lists)
返回一个不可下载的zip对象 。
如果你需要索引访问,你可以使用
by_cols = list(zip(*list_of_lists))
给你一个在两个版本的Python列表的列表。
另一方面,如果你不需要索引访问,你想要的只是build立一个索引列名称的字典,一个zip对象就好了…
file = open('some_data.csv') names = get_names(next(file)) columns = zip(*((x.strip() for x in line.split(',')) for line in file))) d = {} for name, column in zip(names, columns): d[name] = column