截断文件的分组

Question

我有以下代码：

# function to truncate files
def truncate(path):
    file_to_truncate = open(path, "w")
    file_to_truncate.truncate()
    file_to_truncate.close()


# truncate all relevant files so they are empty and new results are not written underneath another set of results
truncate(r'outputA.csv')
truncate(r'\outputB.csv')
truncate(r'outputC.csv')
truncate(r'outputD.csv')
truncate(r'outputE.csv')
truncate(r'MoutputA.csv')
truncate(r'MoutputB.csv')
truncate(r'MoutputC.csv')
truncate(r'MoutputD.csv')
truncate(r"Full Results.csv")
truncate(r'results.csv')

有没有可能缩短这段代码，即截断目录中的所有文件。

Answer 1

您可以使用os.listdir来获取特定目录中的所有文件。然后遍历文件列表，并对它们调用函数：

for file_name in os.listdir():
    truncate(file_name)