我需要同时删除 csv 和 xlsx 文件

Question

我正在将数据提取到 csv 和 xlsx 我需要同时删除这些文件

我的代码：

Csv_files=glob.glob(os.path.join("*.csv))
xl_files=glob.glob(os.path.join("*.xlsx))
for f in csv_files:
 os.remove(f)
for f in xl_files:
 os.remove(f)

我需要删除 3/5 行代码而不是单独删除

Answer 1

itertools 有很多实用方法可以解决这类问题

import itertools
Csv_files=glob.glob(os.path.join("*.csv"))
xl_files=glob.glob(os.path.join("*.xlsx"))

for f in itertools.chain(csv_files, xl_files):
    os.remove(f)

Answer 2

from glob import glob
import os
for f in glob('*.csv') + glob('*.xlsx'):
    os.remove(f)

我需要同时删除 csv 和 xlsx 文件

I need to delete csv an xlsx files simutaneously

python

delete-file