Question

我想知道在以下情况下使用 List Comprehension 避免嵌套循环是个好主意

想象一下config.py文件中的以下名称列表

# config.py
NAME_LIST = [
    'name1',
    'name2',
    (...)
]

我正在使用此名称列表浏览多个不同的目录，对于找到的每个文件，我都会读取文件内容并处理数据。

这是我的实际代码：

import config

def foo():

    for w in config.NAME_LIST:
        folder = "/data/test/{name}/in".format(name=w)
        files = list_files(folder) # return a list of files found in folder

        for f in files:
            content = read_file(folder, f) # read files content
            # (... do some stuff to process data)

我真的很想避免嵌套循环，以使代码尽可能平坦。

我尝试使用 List Comprehension 来解决这个问题，但不确定这是最好的解决方案，因为我不希望它返回任何内容。

这是我所做的：

import config

def foo():

    def _process(f):
        """function called from list comprehension to avoid doing nested loops"""
        content = read_file(folder, f)  # read files content
        # (... do some stuff to process data)

    for w in config.NAME_LIST:
        folder = "/data/test/{name}/in".format(name=w)
        files = list_files(folder) # return a list of files found in folder
        [_process(f) for f in files] # using list comprehension

在这种情况下，列表理解是一个好的解决方案吗？还有更好的方法吗？

Answer 1

有一些使用过滤器的技巧或其他技巧，以避免不必要的输出（Is there a map without result in python?），但是for循环可能是最惯用的方法？您是否有理由要它放平？

使用列表理解

1 个答案: