计算txt文件中数字的平均值

Question

我正在尝试在 python 中编写一个函数，它读取文件，提取文件中冒号后的数字，并返回数字的平均值

（平均值的简单公式 => 数字的总和 / 数字的数量）

文件看起来像：

数字1:7

#more 可以添加

thenumber2: 4（#yes冒号后有空格）

应忽略以“#”开头的行。

到目前为止我的代码：

import os

def get_average_n(path):


s = ""
t = 0
v = 0

if not os.path.exists(path):
    return None

if os.stat(path).st_size == 0:
    return 0.0

else:
    p = open(path, "r")

    content = p.readlines()

    for line in content:
        if line.startswith("#"):
            continue

        elif not line.startswith("#"):
            x = line.find(":")
            s += line[x + 1:]
            h = s.replace("\n", "")

            for c in h:
                if c.isdigit():
                    t += float(c)
                    v += 1
                    avg = round(t / v, 2)

return avg
print(get_average_n("file.txt"))

在 casementionet (fat) 中输出应该是 5.5 但我得到错误的输出，我真的找不到问题。它返回 6.25 而不是 5.5。

Answer 1

为了澄清我的评论。

我修正了缩进，我得到了正确的结果。

ss = '''
thenumber1:7

#more could be added

thenumber2: 4
'''.strip()

with open('file.txt','w') as f: f.write(ss)

#################

import os

def get_average_n(path):
    s = ""
    t = 0
    v = 0

    if not os.path.exists(path):
        return None

    if os.stat(path).st_size == 0:
        return 0.0

    else:
        p = open(path, "r")
        content = p.readlines()

    for line in content:
        if line.startswith("#"):
            continue
        elif not line.startswith("#"):
            x = line.find(":")
            s += line[x + 1:]
            h = s.replace("\n", "")

    for c in h:
        if c.isdigit():
            t += float(c)
            v += 1
            avg = round(t / v, 2)

    return avg
    
print(get_average_n("file.txt"))  # 5.5

------- 为了更简洁地解析文件，试试这个：

def get_average_n(path):
    numlst = []

    if not os.path.exists(path):
        return None

    if os.stat(path).st_size == 0:
        return 0.0

    else:
        with open(path, "r") as p:
           content = p.readlines()

    for line in content:
        if not line.startswith("#") and line.find(":") >= 0:
            numlst.append(float(line.split(':')[1].strip()))

    avg = sum(numlst)/len(numlst)

    return avg

Answer 2

我添加了使用正则表达式的解决方案。我喜欢常规，因为它们具有出色的查找、搜索和替换功能。见下文。

注意：将lines.py & file.txt 复制到目录并运行。你会得到4.75，即(7 + 4 + 3.25) / 3。

参考链接：Python Regex instantly replace groups、How to extract a floating number from a string

» file.txt

thenumber1:7

#more could be added

thenumber2: 4 

thenumber1: 3.25

» lines.py

import os, re

def get_average_n(path):
    if not os.path.exists(path):
        return 0.0

    if os.stat(path).st_size == 0:
        return 0.0
    else:
        total, avg, count = 0.0, 0.0, 0
        regex = r"(.*):\s*(\d+(\.\d+)?)(.*)\s*" # Regex to match interger & 3.25, 4.56 kind of floats
        p = open(path, "r")
        content = p.readlines()

        for line in content:
            if line.strip().startswith('#'):
                continue

            num_s = re.sub(regex, r'\2', line).strip()
            if num_s:
                total += float(num_s)
                count += 1

        if count:
            avg = total / count

    return avg


if __name__ == "__main__":
    print(get_average_n('file.txt'))

» 最后使用python lines.py 运行。

谢谢。