在文件夹中的每个文本文件上运行Python脚本

Question

有没有办法在文件夹中的每个文本文件上运行python脚本？ 所以例如我想删除该文件夹中每个文本文件的前5行我需要做什么？

Answer 1

使用glob模块

简单匹配目录中的所有文件

import glob
import os
import os.path
import sys

dir_of_interest = sys.argv[1]
files = glob.glob(os.path.join(dir_of_interest, "*.txt"))
for f in files:
    ... do stuff ...

Answer 2

这应该有效：

import os
for file in os.listdir("path_to_your_dir"):
    if file.endswith(".txt"):
        #do_simething

Answer 3

详细说明marstrand的回答：

import glob
import os.path
import sys

dir_of_interest = sys.argv[1]
files = glob.glob(os.path.join(dir_of_interest, "*.txt"))
for file in files:
    with open(file, "rw") as f:
         f.writelines(f.readlines()[5:])