从任意文件中抓取数组

Question

为清晰起见进行了编辑。

我知道我们有文件 IO 来访问文件元素并对其进行操作等。我的问题是我很好奇。

假设我们有几个不同的文件，在每个文件中我们可以假设只有一个list 包含此list 中的一些数据。

import os
def make_bars(self):
    for files in os.listdir(os.path.dirname(os.path.abspath(__file__))):
        if files.endswith('.txt'): # or some other file like .py
            a_list = open(files, "r").read()

Answer 1

我对你的问题有点困惑，所以也许我没有解决手头的问题，但你看过os.walk吗？

for root, dirs, files in os.walk(SOME_ROOT_DIR):
    # root is SOME_ROOT_DIR, then each dir in dirs in turn
    # dirs is a list of all the directories in root
    # files is a list of all files in root
    for filepath in filter(lambda s: s.endswith(".py"), files):
        with open(filepath) as f:
            # do something with the .py file

例如这样的目录列表：

C:\USERS\ADSMITH\TMP\ROOT
│   foobar.txt
│   main.py
│   tasks.py
│
└───subroot
        foo.py
        foobar.txt

我能做到：

import os

for root, dirs, files in os.walk("C:/users/adsmith/tmp/root"):
    print("root is " + root)
    for filepath in files:
        if filepath.endswith('.txt') or filepath.startswith('tasks'):
            print(filepath)

# OUTPUT:  NOTE THE RELATIVE PATHS
root is C:/users/adsmith/tmp/root
foobar.txt
tasks.py
root is C:/users/adsmith/tmp/root\subroot
foobar.txt