Question

我正在尝试读取文件并将其内容写入元组列表。每个元组的长度为2，列表的长度根据我们读取的文件而变化。每个元组应代表x，y平面中的一个点，第一个值是x坐标，第二个值是y。我遇到的问题是我认为for循环是执行此操作的最佳方法，但文件中的第一行是表示文件中有多少点的单个值，并且根本不应包含在最终列表中。

def readDataPts(filename):
    """Reads data from an input file and returns a list of tuples
       [(x0,y0), (x1, y1), ...]
    """
    file = open(filename, 'r')
    lines = file.readlines()
    listPts = []
    for line in lines:
        for ptx, pty in line:
            x = (ptx, pty)
        listPts.append(x)
    return listPts

输入的一个例子是：

10
96 571
45 734
174 416
357 259
88 97
188 495
348 443
301 503
719 177
182 237

，输出应为：

[(96, 571), (45, 734), (174, 416), (357, 259), (88, 97).....]

有没有办法从第二行启动for循环？或者有更简单的方法来解决这个问题吗？

由于

Answer 1

您可以在文件对象上调用 next 以跳过第一行并从第二行开始然后分割每一行并调用元组或让 csv.reader 解析每一行并映射到元组：

分裂：

with open("in.txt") as f:
    next(f) # skip first line
    arr = [tuple(line.split()) for line in f]
    print(arr)

csv lib：

import  csv
with open("in.txt") as f:
    next(f) # skip first line
    arr = list(map(tuple, csv.reader(f,delimiter=" ")))
    print(arr)

两者都将返回：

[('96', '571'), ('45', '734'), ('174', '416'), ('357', '259'), ('88', '97'), ('188', '495'), ('348', '443'), ('301', '503'), ('719', '177'), ('182', '237')]

如果你想要整理：

with open("in.txt") as f:
    next(f) # skip first line
    arr = [tuple(map(int, line.split())) for line in f]
    print(arr)

和csv.reader：

import  csv
with open("in.txt") as f:
    next(f) # skip first line
    arr = [tuple(map(int,row) for row in  csv.reader(f,delimiter=" "))]
    print(arr)

哪个会给你：

 [(96, 571), (45, 734), (174, 416), (357, 259), (88, 97), (188, 495), (348, 443), (301, 503), (719, 177), (182, 237

Answer 2

您可以使用.split()从每行创建元组并检查长度：

def readDataPts(filename):
    listPts = []
    with open(filename, 'r') as f:
        for line in f:
            numbers = line.split()
            if len(numbers) == 2:
                listPts.append(map(int, numbers))
    return listPts

Answer 3

你可以使用告诉循环只遍历行[1：]，这将跳过第一行并使用所有其余行。

def readDataPts(filename):
"""Reads data from an input file and returns a list of tuples
   [(x0,y0), (x1, y1), ...]
"""

    file = open(filename, 'r')
    lines = file.readlines()
    listPts = []
    for line in lines[1:]:
        for ptx, pty in line:
            x = (ptx, pty)
        listPts.append(x)
    return listPts

Answer 4

只是另一个想法。

with open('in.txt') as f:
    nums = map(int, f.read().split())
    print zip(nums[1::2], nums[2::2])

将文件写入元组列表

4 个答案: