python - 如何计算文件中文本文件中重复的值之间的行数

Question

我想计算在以下示例中 'A' 之后和 'A' 重复之前的文本文件的行数:

A1
   Apple
   Mangoes
   Oranges
A2
   Strawberries
   Grapefruit
A3
   Papaya 
   Sweet_lime

我希望结果是这样的

A1: 3, A2: 2, A3: 2

到目前为止我已经尝试过:

f = open("temp.txt","r")
count = 0
A_vals = []
for lines in f:
    value = re.search("(A\d)",lines)
    header = value.group(1)
    if header in lines:
        count += 1
    A_vals.append(count)

我没有得到预期的结果。任何帮助表示赞赏!

Answer 1

由于您有 A 和数字，因此请使用正则表达式来匹配模式。您可以使用默认字典作为计数器。

import re
from collections import defaultdict

A_vals = defaultdict(int)
current_A = None

with open('temp.txt') as fp:
    for line in fp:
        m = re.match(r'A\d+', line)
        if m:
            current_A = m.group()
            continue
        if current_A:
            A_vals[current_A] += 1

print(dict(A_vals))
# prints:
{'A1': 3, 'A2': 2, 'A3': 2}