Python-从文本文件中读取数字并放入列表

Question

因此，就像标题中所说的，我开始学习一些python，而我在使用这项技术时遇到了麻烦。 我需要完成的工作是读一些数字并将它们存储在列表中。 文本文件如下所示：

0 0 3 50

50 100 4 20

基本上，这些是用于python的海龟制作形状的坐标和方向。 我要讲的是，唯一的问题是使它们采用正确的格式。 所以我不知道如何将这些数字从文件中获取到[ [0, 0, 3, 50], [50, 100, 4, 20] ]列表中，每个四个坐标是一个列表一个大清单。

这是我的尝试，但正如我所说，我需要一些帮助-谢谢。

polyShape=[]
infile = open(name,"r")
num = int(infile.readline(2))
while num != "":
    polyShape.append(num)
    num = int(infile.readline(2))
infile.close()

Answer 1

with open('data.txt') as f:
    polyShape = []
    for line in f:
        line = line.split() # to deal with blank 
        if line:            # lines (ie skip them)
            line = [int(i) for i in line]
            polyShape.append(line)

会给你

[[0, 0, 3, 50], [50, 100, 4, 20]]

这将适用于包含（或不包含）空行的文件。

完成操作或遇到异常时，使用with构造将自动为您关闭文件。

Answer 2

假设您的输入文件中实际上没有空行：

with open(name, "r") as infile:
    polyShape = [map(int, line.split()) for line in infile]

说明： map(int, line.split())分割每一line并将每一部分转换为一个int 。 [X for Y in Z]构造是一个列表推导，它依次将映射map到文件的所有行上，并将其结果返回到列表中。

如果您现在觉得这太复杂了，则map(int, line.split())是主要的带回家的消息。

Answer 3

with open('data.txt') as f:
    lis=[map(int,x.split()) for x in f if x.strip()]   # if x.strip() to skip blank lines

   #use list(map(int,x.split()))  in case of python 3.x

这就是map()工作方式：

>>> map(int,'1 2 3 4'.split())
[1, 2, 3, 4]

Answer 4

遍历文件是最简单的方法：

poly_shape = []

with open(name, 'r') as handle:
    for line in handle:
        if not line.strip():
            continue  # This skips blank lines

        values = map(int, line.split())
        poly_shape.append(values)

Answer 5

单线：

[ [int(x) for x in line.split(' ')] for line in open(name,'r').readlines() if line.strip()]

但readlines部分可能不是一个好主意。

我非常确定[int(x) for x in ... ]比其他建议的解决方案中使用map更快。

编辑

感谢Blender：不需要.readlines ，这很酷，所以我们只有：

[ map(int, line.split()) for line in open(name,'r') if line.strip()]

我还使用了map(int, )因为它实际上更快，并且您也可以只使用line.split()代替line.split(' ') 。

Answer 6

我不建议对大数组使用append 。 这比创建零数组和为其赋值要慢50倍。

import numpy
fname = "D:\Test.txt";
num_lines = sum(1 for line in open(fname));
array = numpy.zeros((num_lines,4));
k = 0;
with open(fname, "r") as ins:
    for line in ins:
        a =[int(i) for i in line.split(' ')];;
         array[k,0:4] =a;
         k = k+1;
print(array)