时间戳

首页 > 解决方案 > 从特定行读取 csv 文件

python - 从特定行读取 csv 文件

问题描述

我试图在找到“X”时提取数据，并从该行到最后获取所有内容。我的代码只给了我下面的行到最后。有没有办法用“X”包含行？

a = []
with open(file_path,'rU') as fp:
        for i,line in enumerate(fp):
            if "X" in line:
                #print (i, line)
                for line in fp:
                    a.append(line.split(' '))
                    print(a)

标签： python

解决方案

鉴于：

$ cat file.csv
line 1
line 2
line 3
line X
line 5
line 6
line 7

您可以使用itertools dropwhile：

import itertools as it

with open(file_path,'rU') as fp:
    a=[line.strip() for line in it.dropwhile(lambda l: 'X' not in l, fp)]

结果：

>>> a
['line X', 'line 5', 'line 6', 'line 7']

如果你想enumerate在整个文件上使用，你可以有一个对象在测试后保持一个True或False状态。通常称为 FlipFlop，它对于在文件中查找标记而仅逐行读取文件很有用：

class FlipFlop: 
    def __init__(self, start_state, func):
        self.call=func
        self.state = start_state
    def __call__(self, st):
        if self.call(st): self.state=not self.state
        return self.state

with open(fn) as f_in:
    ff=FlipFlop(False, lambda s: 'X' in s)
    x_and_after=[(i, line.strip()) for i, line in enumerate(f_in) if ff(line)]

结果：

>>> x_and_after
[(3, 'line X'), (4, 'line 5'), (5, 'line 6'), (6, 'line 7')]

推荐阅读