Python - Чтение файла из и в определенные строки текста
Я не говорю о конкретных номерах строк, потому что я читаю несколько файлов с одинаковым форматом, но различаюсь по длине.
Скажем, у меня есть этот текстовый файл:
Something here...
... ... ...
Start #I want this block of text
a b c d e f g
h i j k l m n
End #until this line of the file
something here...
... ... ...
Надеюсь, ты знаешь, что я имею в виду. я думал об итерации через файл, затем, используя регулярное выражение, чтобы найти номер строки "Пуск" и "Конец", затем используйте linecache для чтения из строки "Пуск" в конец строки.
Но как получить номер строки? какую функцию я могу использовать?
Ответы
Ответ 1
Если вам просто нужен блок текста между Start и End, вы можете сделать что-то простое, например:
with open('test.txt') as input_data:
# Skips text before the beginning of the interesting block:
for line in input_data:
if line.strip() == 'Start': # Or whatever test is needed
break
# Reads text until the end of the block:
for line in input_data: # This keeps reading the file
if line.strip() == 'End':
break
print line # Line is extracted (or block_of_lines.append(line), etc.)
На самом деле вам не нужно манипулировать номерами строк, чтобы читать данные между маркерами Start и End.
Логика ( "читать до..." ) повторяется в обоих блоках, но она достаточно ясна и эффективна (другие методы обычно включают проверку некоторого состояния [до того, как блок/внутри блока/конца достигнутого блока], что наступает за время пени).
Ответ 2
Здесь что-то будет работать:
data_file = open("test.txt")
block = ""
found = False
for line in data_file:
if found:
block += line
if line.strip() == "End": break
else:
if line.strip() == "Start":
found = True
block = "Start"
data_file.close()
Ответ 3
Вы можете легко использовать регулярное выражение. Вы можете сделать его более надежным по мере необходимости, ниже - простой пример.
>>> import re
>>> START = "some"
>>> END = "Hello"
>>> test = "this is some\nsample text\nthat has the\nwords Hello World\n"
>>> m = re.compile(r'%s.*?%s' % (START,END),re.S)
>>> m.search(test).group(0)
'some\nsample text\nthat has the\nwords Hello'
Ответ 4
Это должно начаться для вас:
started = False
collected_lines = []
with open(path, "r") as fp:
for i, line in enumerate(fp.readlines()):
if line.rstrip() == "Start":
started = True
print "started at line", i # counts from zero !
continue
if started and line.rstrip()=="End":
print "end at line", i
break
# process line
collected_lines.append(line.rstrip())
Генератор enumerate
принимает генератор и перечисляет итерации.
Например.
print list(enumerate("a b c".split()))
печатает
[ (0, "a"), (1,"b"), (2, "c") ]
UPDATE
плакат попросил использовать регулярное выражение для соответствия строкам типа "===" и "======":
import re
print re.match("^=+$", "===") is not None
print re.match("^=+$", "======") is not None
print re.match("^=+$", "=") is not None
print re.match("^=+$", "=abc") is not None
print re.match("^=+$", "abc=") is not None