209

我正在尝试将文本文件的行读入 python 中的列表或数组。我只需要能够在创建列表或数组后单独访问它。

文本文件格式如下:

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

显然,它将整个文件读入仅包含一项的列表,而不是单个项目的列表。我究竟做错了什么?

4

6 回答 6

174

您必须使用将字符串拆分为值列表split()

所以,

lines = text_file.read().split(',')

编辑:我没有意识到这会有这么大的吸引力。这是一种更惯用的方法。

import csv
with open('filename.csv', 'r') as fd:
    reader = csv.reader(fd)
    for row in reader:
        # do something
于 2013-02-03T19:29:10.763 回答
61

您还可以使用 numpy loadtxt 之类的

from numpy import loadtxt
lines = loadtxt("filename.dat", comments="#", delimiter=",", unpack=False)
于 2013-02-04T10:46:55.860 回答
25

所以你想创建一个列表列表......我们需要从一个空列表开始

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 版本中的列表列表。

另一方面,如果您不需要索引访问并且您想要的只是构建一个按列名索引的字典,那么一个 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
于 2014-11-20T17:47:24.583 回答
8

这个问题是询问如何将文件中的逗号分隔值内容读取到可迭代列表中:

0,0,200,0,53,1,0,255,...,0.

最简单的方法是使用csv模块,如下所示:

import csv
with open('filename.dat', newline='') as csvfile:
    spamreader = csv.reader(csvfile, delimiter=',')

现在,您可以像这样轻松地迭代spamreader

for row in spamreader:
    print(', '.join(row))

有关更多示例,请参阅文档

于 2018-03-14T05:36:30.273 回答
0

我有点晚了,但您也可以将文本文件读入数据框,然后将相应的列转换为列表。

lista=pd.read_csv('path_to_textfile.txt', sep=",", header=None)[0].tolist() 

例子。

lista=pd.read_csv('data/holdout.txt',sep=',',header=None)[0].tolist()

注意:相应数据框的列名将采用整数形式,我选择 0 因为我只提取第一列

于 2021-09-24T13:27:06.330 回答
0

这样更好,

 def txt_to_lst(file_path):

    try:
        stopword=open(file_path,"r")
        lines = stopword.read().split('\n')
        print(lines)

    except Exception as e:
        print(e)
于 2021-10-21T07:39:22.107 回答