Question

如果目录中存在any CSV文件，我试图找到一种编写脚本的方法，请将其删除。如何在不指定特定文件名的情况下编写此代码？

如果文件扩展名为.py，是否可以写，用os将其删除？

import os

if os.path.exists("AHU4_data_CSV.csv"):
  os.remove("AHU4_data_CSV.csv")

else:
  print("The file does not exist")

Answer 1

您可以执行以下操作：

import os

root = "e:/"

for root, dirs, files in os.walk(root):
    for file in files:
        if file.endswith('.csv'):
            print("removed: " + os.path.join(root, file) )
            os.remove( os.path.join(root, file) )

它将从E中删除您的E目录中的所有CSV文件，包括子文件夹的CSV文件。

Answer 2

这应该工作-只需列出目录中的所有文件/文件夹，如果它们不以.py结尾，则将其删除。

for filename in os.listdir('.'):
    if not filename.endswith('.py'):
        if os.path.isfile(filename):
            os.remove(filename)
        else:
            os.rmdir(filename)

Answer 3

你可以这样做

import os
mydir='path/to/dir/with/files'

for file in os.listdir(mydir):
    if file.endswith('.csv'):
        os.remove(os.path.join(mydir,file))

操作系统从目录中删除.py文件以外的任何文件

3 个答案: