如何使用python打开一些特定的文件

Question

我有几个名为yyyymmdd-hhmmss.csv.gz 的文件。现在我想从凌晨 2 点开始打开每天的第一个文件，一个一个地做一些处理。

我的文件是：

...
20120706-015923.txt
20120706-015934.txt 
20120706-020008.txt
20120706-020030.txt
...
20120707-015934.txt
20120707-020005.txt
20120707-020015.txt
...

所以我想先查看这个文件夹中的文件名，找到20120706-020008.txt，这是当天凌晨2点开始的第一个文件，然后我有一个脚本来读取所有行并进行一些处理。我的问题是，如何告诉这个脚本找到这些文件？ 20120706-020008.txt, 20120707-020015.txt, 20120708-02....

Answer 1

for loop和dictionary的一些简单用法：

import glob
import datetime

dic={}

for f in glob.glob('*-*.txt'):
    d = datetime.datetime.strptime(f, '%Y%m%d-%H%M%S.txt')
    if d.hour < 2:
        continue
    if not d.day in dic or dic[d.day][0] > d.time:
        dic[d.day] = (d.time, f)

for t, f in dic.values():
    print f

---

鉴于以下文件：

20120706-015923.txt
20120706-015934.txt 
20120706-020008.txt
20120706-020030.txt
20120707-015934.txt
20120707-020005.txt
20120707-020015.txt

输出将是：

20120706-020008.txt
20120707-020005.txt

Answer 2

您可以使用glob 模块来遍历文件：

import glob
for f in glob.glob('20120707-*'):
    # do something with f

上面的代码意思是：遍历所有文件名以20120707-开头的文件。 * 符号与文件名的其余部分匹配。

您可以使用 datetime 模块在 Python 代码中计算文件的前缀（请参阅 documentation 以了解您可以为 strftime 使用哪些值）：

import datetime
today = datetime.date.today()
print today.strftime("%Y%m%d")

结合代码变成：

import datetime    
import glob

today = datetime.date.today()
pattern = "{0}-*".format(today.strftime("%Y%m%d"))
for f in glob.glob(pattern):
    # do something with f

您还可以强制 glob 只查找 .txt 文件，方法是将其包含在搜索模式中：

pattern = "{0}-*.txt".format(today.strftime("%Y%m%d"))

Answer 3

您希望将文件按时间顺序分组到凌晨 2 点到凌晨 2 点之间的时间，对它们进行排序并获取第一个：

import itertools as it
import glob
import datetime

def map_to_day_2am(fn):
    # parses filename, substracts two hours and returns the date
    return (datetime.strptime(fn, '%Y%m%d-%H%M%S.txt') -
            datetime.timedelta(hours=2)).date()

for day, daily_fns_iter in it.groupby(sorted(glob.glob('*.txt')),
                                      key=map_to_day_2am):
    print min(daily_fns_iter)