有没有办法在字符串的特定范围内使用“for循环”？

Question

你好 stackoverflow 用户，

我想知道如何将 for 循环与字符串一起使用。

例如，

有一个文件（file.txt）之类的，

======================

Initial Value

    1 2 3
    3 4 5
    5 6 7

Middle Value           <---From Here

    3 5 6
    5 8 8
    6 9 8                  <---To Here

Last Value

    5 8 7
    6 8 7
    5 5 7

===================

我想只在“中值”中修改文件的部分并写一个输出文件

修改后。

我认为如果我使用“if and for”语句，这可能会得到解决。

我想到了类似的代码

with open('file.txt') as f, open('out.txt', 'w') as f2:

       for line in f:
           sp1 = line.split()
           line = " ".join(sp1) + '\n'
           if line == 'Middle':
              "Do something until line == 'Last'"

我被"Do something until line == 'Last'" 卡住了。

感谢任何 cmets。

谢谢。

Answer 1

共有三种基本方法。

---

首先是使用状态机。您可以构建一个真实的状态机，但在这种情况下，状态和转换非常简单，只需使用一个标志就可以更简单地伪造它：

state = 0
for line in f:
    sp1 = line.split()
    line = " ".join(sp1) + '\n'
    if state == 0:
        if line == 'Middle\n':
            state = 1
    elif state == 1:
        if line == 'Last\n':
            state = 2
        else:
            # Thing you do until line == 'Last\n'
    else:
        # nothing to do after Last, so you could leave it out

请注意，我检查的是'Middle\n'，而不是'Middle'。如果您查看上面构建line 的方式，它无法匹配后者，因为您总是添加'\n'。但还要注意，在您的示例数据中，该行是'Middle Value\n'，而不是'Middle'，所以如果在您的真实数据中是这样，您必须在此处处理。是line == 'Middle Value\n'、line.startswith('Middle') 还是其他，取决于您的实际数据，只有您自己知道。

---

或者，您可以将其分解为循环：

for line in f:
    sp1 = line.split()
    line = " ".join(sp1) + '\n'
    if line == 'Middle\n':
        break
for line in f:
    sp1 = line.split()
    line = " ".join(sp1) + '\n'
    if line == 'Last\n':
        break
    else:
        # Thing you do until line == 'Last\n'
for line in f:
    # Nothing to do here, so you could leave the loop out

这个也有变化。例如：

lines = (" ".join(line.split()) + '\n' for line in f)
lines = dropwhile(lambda line: line != 'Middle', lines)
middle = takewhile(lambda line: line != 'End', lines)
for line in middle:
    # Thing you want to do

---

最后，您可以在之前将文件拆分为行，而不是之后。这更难迭代执行，所以让我们将整个文件读入内存来展示这个想法：

contents = f.read()
_, _, rest = contents.partition('\nMiddle\n')
middle, _, _ = rest.partition('\nEnd')
for line in middle.splitlines():
    # Thing you want to do

如果将整个文件读入内存会浪费太多空间或需要很长时间才能开始，mmap 是你的朋友。

Answer 2

我会将这个过程编码为一个简单的 FSM（Finite-State Machine 或更具体地说是 event-driven Finite-state machine）：

with open('file.txt') as f, open('out.txt', 'w') as f2:
    state = 1
    for line in f:
        if line == 'Middle Value\n':
            state = 2
            continue  # unless there's something to do upon entering the state
        elif line == 'Last Value\n':  # might want to just test for blank line `\n'
            state = 3
            continue  # unless there's something to do upon entering the state

        # otherwise process to line based on the current value of "state"
        if state == 1:      # before 'Middle Value' has been seen
            pass
        elif state == 2:    # after 'Middle Value' has been seen
            pass
        else:               # after 'Last Value' (or a blank line after 
            pass            # 'Middle Value') has been seen

只需将pass 语句替换为在读取输入文件时可以执行的任何操作即可。

Answer 3

在您的 if line == 'Middle': 中，您可以翻转一个布尔标志，允许您输入另一个 if inMiddle and line !=last` 语句，然后您可以在其中修改您的数字

你可以用这个替换你的 for 循环。

inMiddle = false
for line in f:
    sp1 = line.split()
    line = "".join(sp1) + '\n'

    if line == 'Middle':
        inMiddle = true

    if inMiddle and line != 'Last':
        #MODIFY YOUR NUMBERS HERE
    elif line == 'Last':
        inMiddle = false

Answer 4

请原谅我访问文件的方式有点不同

with open('file.txt') as f:
    file_string = f.read()

middle_to_end = file_string.split('Middle Value\n')[-1]
just_middle   = middle_to_end.split('Last Value\n')[0]

middle_lines = just_middle.splitlines()
for line in middle_lines:
    do_something

Answer 5

基本上，您正在设置一个标志来表示您在“该部分”中。下面我可以选择在完成时设置一个不同的标志。例如，您可以在 flag 为 2 时退出。

with open('file.txt') as f, open('out.txt', 'w') as f2:

   section = 0;

   for line in f:
      if line.startswith("Middle"):
          section = 1
      elif line.startswith("Last"):
          section = 2



      if section == 1:
        #collect digits and output to other file     
        f2.write(line)         

      elif section == 2:
         #close file and break out
         f.close()
         f2.close()
         break
      else:
          continue