如果文件位于同一文件夹中，请将它们相互添加

Question

我有txt文件，包含不同文件夹中的数字。它们将被转换为numpy数组并相互连接。我有以下脚本：

import numpy as np
import os 

main_directory = "mypath"
for root, dirs, files in os.walk(main_directory):
     for name in files:
          if name.endswith(".txt"):
              os.chdir(root)
              ff = os.path.join(root, name)
              text = open(str(ff)).read()
              a = np.array(text.split("\n").split(), dtype=float)

假设a，b和c是同一文件夹中的文本文件。我想为每个文件夹做这样的事情：

final = my_function(numpy.concatenate([a, b, c])

请注意，我不知道要合并的文本文件数。