每次在python中再次运行代码之前删除文件

Question

我有一个程序，比如check.py，要运行这个程序，我需要5-6个支持文件，比如1.py，2.data，3.test，4.train，5.txt。在我使用python3 check.py运行该代码之后，将创建60个新文件作为输出。因此，在再次运行check.py之前，我希望使用python脚本清除以前创建的所有60个文件。

我想要这样做的步骤是，我想在该文件夹中创建一个所需文件的列表。例如，在上述情况下，这些例子是：1.py、2.data、3.test、4.train、5.txt和check.py。然后，我想删除所有剩余的文件，这些文件不在此列表中。我能这样做吗？请帮帮我。

Answer 1

我会在子文件夹中创建新的文件，然后在不再感兴趣的时候删除该文件。就不会那么乱了。否则，os.listdir()将给出目录中所有文件的列表。只需要移除你想要保留的那些。