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

Question

有没有办法让一个Python脚本在文件夹里的每个文本文件上运行呢？比如说，我想删除这个文件夹里每个文本文件的前5行，我该怎么做呢？

Answer 1

为了更详细地解释一下tmarstrand的回答：

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:])

Answer 2

这个应该可以正常运行：

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

Answer 3

简单来说，可以使用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 ...