为特定文件扩展名搜索目录并执行求和以将其内容添加在一起的后续操作

Question

这是对我的previous post 的跟进。本质上，这个想法是将所有以 .math 结尾的文件的内容添加到 /home/pi/math 中。目录。目前我有 3 个文件：test.math、hello.math、hi.math - 它们都至少有 1.00。问题在于 1. 它似乎没有正确读取文件的内容，也没有将它们加起来。

这个想法是，它将添加这 3 个文件（最终我想将它们添加到一个变量中）并显示总数。请注意，我确实想包含浮点数。

import os

srcdir = "/home/pi/math/"
files = os.listdir(srcdir)


def sum_file_contents():
    value = float()
    for file in srcdir:
      if file.endswith(('math')):
          value += float(open(file, "r").read().strip())
      return value

for file in files:
    if file.endswith(('math')):
       print(file)
       print(sum_file_contents())

输出

test.math
0.0
hello.math
0.0
hi.math
0.0

Answer 1

您的主要for 循环遍历来自os.listdir(srcdir) 的files，但随后您的sum_file_contents() 尝试再次执行相同的操作，除了它以字符串的形式遍历srcdir，这很自然不给你目录列表。主要的for 循环是多余的，您应该通过在迭代它时调用os.listdir() 来简单地修复sum_file_contents()：

import os

srcdir = "/home/pi/math/"

def sum_file_contents():
    value = float()
    for file in os.listdir(srcdir):
      if file.endswith(('math')):
          value += float(open(os.path.join(srcdir, file), "r").read().strip())
    return value
print(sum_file_contents())

Answer 2

import os
srcdir = "/home/pi/math/"
files = os.listdir(srcdir)

def sum_file_contents(file):
    value = 0.0
    f = open(file,'r')
    for line in f:
        value += float(line.strip())
    return value

total_value = 0.0
for file in files:
    if file.endswith(('math')):
        print(file)
        print(sum_file_contents(file))
        total_value += sum_file_contents(file)
print(total_value)

即使文件中有更多带有浮点数的行，这也会处理。