根据月份和年份算术生成文件名列表答案

【问题标题】：Generate a list of file names based on month and year arithmetic根据月份和年份算术生成文件名列表
【发布时间】：2011-02-14 04:19:32
【问题描述】：

如何列出数字 01 到 12（12 个月中的每个数字），以便当前月份总是排在最后，而最旧的月份排在最后。换句话说，如果该数字大于当前月份，则为上一年。

例如02 是 2011 年 2 月（当前月份），03 是 2010 年 3 月，09 是 2010 年 9 月，但 01 是 2011 年 1 月。在这种情况下，我想要 @ 987654327@。这就是我要确定年份的方法：

for inFile in os.listdir('.'):
    if inFile.isdigit():
    month = months[int(inFile)]
       if int(inFile) <= int(strftime("%m")):
           year = strftime("%Y")
       else:
           year = int(strftime("%Y"))-1
       mnYear = month + ", " + str(year)

我不知道下一步该做什么。我应该在这里做什么？

更新：

我认为，我最好上传整个脚本以便更好地理解。

#!/usr/bin/env python

import os, sys
from time import strftime
from calendar import month_abbr

vGroup = {}
vo = "group_lhcb"
SI00_fig = float(2.478)
months = tuple(month_abbr)

print "\n%-12s\t%10s\t%8s\t%10s" % ('VOs','CPU-time','CPU-time','kSI2K-hrs')
print "%-12s\t%10s\t%8s\t%10s" % ('','(in Sec)','(in Hrs)','(*2.478)')
print "=" * 58

for inFile in os.listdir('.'):
    if inFile.isdigit():
        readFile = open(inFile, 'r')
        lines = readFile.readlines()
        readFile.close()

        month = months[int(inFile)]
        if int(inFile) <= int(strftime("%m")):
            year = strftime("%Y")
        else:
            year = int(strftime("%Y"))-1
        mnYear = month + ", " + str(year)

        for line in lines[2:]:
            if line.find(vo)==0:
                g, i = line.split()
                s = vGroup.get(g, 0)
                vGroup[g] = s + int(i)

        sumHrs = ((vGroup[g]/60)/60)
        sumSi2k = sumHrs*SI00_fig
        print "%-12s\t%10s\t%8s\t%10.2f" % (mnYear,vGroup[g],sumHrs,sumSi2k)
        del vGroup[g]

当我运行脚本时，我得到了这个：

[root@serv07 usage]# ./test.py 

VOs               CPU-time  CPU-time     kSI2K-hrs
                  (in Sec)  (in Hrs)      (*2.478)
==================================================
Jan, 2011        211201372     58667     145376.83
Dec, 2010          5064337      1406       3484.07
Feb, 2011         17506049      4862      12048.04
Sep, 2010        210874275     58576     145151.33

正如我在原帖中所说，我希望结果是这样的：

Sep, 2010        210874275     58576     145151.33
Dec, 2010          5064337      1406       3484.07
Jan, 2011        211201372     58667     145376.83
Feb, 2011         17506049      4862      12048.04

源目录中的文件如下所示：

[root@serv07 usage]# ls -l
total 3632
-rw-r--r--  1 root root 1144972 Feb  9 19:23 01
-rw-r--r--  1 root root  556630 Feb 13 09:11 02
-rw-r--r--  1 root root  443782 Feb 11 17:23 02.bak
-rw-r--r--  1 root root 1144556 Feb 14 09:30 09
-rw-r--r--  1 root root  370822 Feb  9 19:24 12

我现在有更好的图片吗？抱歉，一开始不是很清楚。干杯！！

更新@Mark Ransom

这是马克建议的结果：

[root@serv07 usage]# ./test.py 

VOs               CPU-time  CPU-time     kSI2K-hrs
                  (in Sec)  (in Hrs)      (*2.478)
==========================================================
Dec, 2010          5064337      1406       3484.07
Sep, 2010        210874275     58576     145151.33
Feb, 2011         17506049      4862      12048.04
Jan, 2011        211201372     58667     145376.83

正如我之前所说，我正在寻找按以下顺序打印的结果：2010 年 9 月 -> 2010 年 12 月 -> 2011 年 1 月 -> 2011 年 2 月 干杯！！

【问题讨论】：

从这里看来您的文件是按月份命名的？
根据您的描述，所需的输出不是 [03, 09, 01, 02] 而不是 [09, 03, 01, 02] 吗？
@geekGod：你是对的 - 它应该是 [03, 09, 01, 02]。这是我的错。
@KyleWpppd：是的，文件以月份命名。这些是一年中每月的 CPU 使用率，我正在尝试按月生成报告。

标签： python

【解决方案1】：

除非您对列表进行排序，否则您无法打印已排序的列表！我认为修改代码的最简单方法是首先获取文件名，对它们进行排序，然后对排序列表进行操作。通过将列表构建为按年月顺序排列的元组，排序将自动为您提供正确的顺序。

filelist = []
for inFile in os.listdir('.'):
    if inFile.isdigit():
        if int(inFile) <= int(strftime("%m")):
            year = strftime("%Y")
        else:
            year = int(strftime("%Y"))-1
        filelist.append((year, inFile))
filelist.sort()
for year, inFile in filelist:
    month = months[int(inFile)]
    ...

【讨论】：

感谢您的帮助马克。它几乎做到了，但不完全是我想要的。我已经用结果更新了我的原始帖子。干杯！！
@MacUsers：简单的错误，它是按字母顺序而不是按数字排序。我现在正在编辑我的答案。

【解决方案2】：

第三次更新：

import os, sys
from time import strftime
from calendar import month_abbr

thismonth = int(strftime("%m"))
sortedmonths = ["%02d" % (m % 12 + 1) for m in range(thismonth, thismonth + 12)]
inFiles = [m for m in sortedmonths if m in os.listdir('.')]

for inFile in inFiles:
    readFile = open(inFile, 'r')
    # [ ... everything else is the same from here (but reindented)... ]

列表推导更可取，但如果我所做的某些事情在 2.3 中不符合规定，请尝试以下操作：

inFiles = filter(lambda x: x.isdigit(), os.listdir('.'))
sortedmonths = map(lambda x: "%02d" % (x % 12 + 1), range(thismonth, thismonth + 12))
inFiles = filter(lambda x: x in inFiles, sortedmonths)

第二次更新：

好的，根据您的编辑，这是我认为最简单的解决方案：

import os, sys
from time import strftime
from calendar import month_abbr

thismonth = int(strftime("%m"))
inFiles = []
for inFile in os.listdir('.'):
    if inFile.isdigit():
        inFiles.append(inFile)
inFiles.sort(key=lambda x: (int(x) - thismonth - 1) % 12)

for inFile in inFiles:
    readFile = open(inFile, 'r')
    # [ ... everything else is the same from here (but reindented)... ]

这样你的主循环会以正确的顺序遍历文件。

你也可以用一行推导替换上面的四行循环：

inFiles = [inFile for inFile in os.listdir('.') if inFile.isdigit()]

另外，我建议使用from datetime import datetime，然后使用datetime.now()。它的界面比strftime() 更好——datetime.now().month 返回数字月份（.year、.min、.second 等相同，str(datetime.now()) 提供格式良好的日期时间字符串。

更新：

好的，经过一番折腾，这是我认为最有效的方法——无论是Jan = 1 还是Jan = 0，它都有效：

>>> themonths = [1, 2, 3, 9]
>>> themonths.sort(key=lambda x: (x - thismonth - 1) % 12)
>>> themonths
[2, 3, 9, 1]

原帖：

如果我理解正确的话：

>>> thismonth = 1
>>> [m % 12 for m in range(thismonth + 1, thismonth + 13)]
[2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 0, 1]

或者如果你只想要几个月，你可以这样做：

>>> thismonth = 1
>>> themonths = [0, 1, 2, 8]
>>> [m % 12 for m in range(thismonth + 1, thismonth + 13) if m % 12 in themonths]
[2, 8, 0, 1]

请注意，在此方案中 Jan = 0 .... Dec = 11。如果你想要Jan = 1，你可以在结果数字上加1（即[m % 12 + 1 for ... if m % 12 + 1 in themonths]）。但我认为Jan = 0 是一个更好的系统，至少对于后端而言。

【讨论】：

@senderle：我认为，您的方法 1 和/或 3 最适合我。但是，我已经更新了我的原件，让你们更好地理解我真正想要做的事情。干杯！！
@senderle：我运气不好，我收到TypeError: sort() takes no keyword arguments 错误。我正在使用 v2.3 - 是问题吗？由于某些其他原因，我必须在这个特定的盒子上坚持使用 v2.3，否则很多事情都会失败。
哇，我没有意识到 key 关键字比 2.3 更新。我没有足够的资金来设置 2.3 测试安装，所以如果我的新解决方案（基于 3 而不是 1）不起作用，请告诉我。
@senderle：是的，key 是在 v2.4 中引入的 - 根据发行说明。无论如何，这一次（你的第三次更新）就像一个魅力；非常感谢你。非常感谢。
@senderle：感谢您提供的信息。我仍在学习正确使用stackoverflow。干杯！！

【解决方案3】：

你的问题我很难理解。

如果您只想轮换月份的 12 位数字，以便当前月份的数字是最后一位，可以这样做：

>>> def rot_list(l,n):
...    return l[n:]+l[:n]
... 
>>> rot_list(range(1,13),2)
[3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 1, 2]

如果您要查找的目标列表（在您的示例中）最终经过排序和旋转，以便最接近当前月份的列表（由 n 表示）位于末尾，则可以这样做：

>>> tar=[1,2,3,9]
>>> n=2
>>> [x for x in range(1,13)[n:]+range(1,13)[:n] if x in tar]
[3, 9, 1, 2]
>>> tar=[3,1,2,9]
>>> [x for x in range(1,13)[n:]+range(1,13)[:n] if x in tar]
[3, 9, 1, 2]

现在，如果您需要该列表具有前导零并且各个元素是字符串：

>>> nm=[3,9,1,2]
>>> fn=['%02i' % x for x in nm]
>>> fn
['03', '09', '01', '02']

无论 Jan 是“0”还是“1”，所有这些方法都有效。只需根据约定将range(1,13) 的任何引用更改为range(0,12)。

编辑

如果我理解你在做什么，这段代码会有所帮助：

import datetime

def prev_month(year,month,day):
    l=range(1,13)
    l=l[month:]+l[:month]
    p_month=l[10]
    if p_month>month:
        p_year=year-1
    else:
        p_year=year
    return (p_year,p_month,1)

def next_month(year,month,day):
    l=range(1,13)
    l=l[month:]+l[:month]
    p_month=l[0]
    if p_month<month:
        p_year=year+1
    else:
        p_year=year
    return (p_year,p_month,1)

year=2011
month=2
t=(year-1,month,1)
ds=[]
for i in range(1,13):
    print datetime.date(*t).strftime('%m, %y')
    t=next_month(*t)

输出：

只要将文件名的适用格式放在strftime()方法中即可...

希望这就是你要找的……

【讨论】：

在月份算术中使用模数而不是列表可能更好，不是吗？