使用python处理文件

背景：

260个学生，分52组，每组1-6人，交作业时只需要一个人代表整个team交，有多次作业，每次可以由不同的人提交。

文件名格式为：学号_姓名。我只负责其中9组，所以要从52个组内挑出我负责的9组，每次对着40多个人的名单从50多个文件名中检索出其中9个，太麻烦。

于是想用脚本挑出我自己负责的那部分，可是又不会windows下的shell，其实linux下也不会。。。。

刚好学了python，就打算用python处理一下。

代码如下：

import osname = "homework2" ###select files in which dir，需要处理的文件所在文件夹####我负责的9个组的学号，每行一个teamtasks=(111250116,111250158,111250196,111250198,111250222,       111250199,111250200,111250210,111250211,128351018,       111250114,111250206,111250213,111250242,111250244,       111250153,111250171,111250214,101250032,101250038,       111250184,111250189,111250205,111250215,111250223,       111250207,111250212,111250216,111250217,111250232,       111250209,111250218,111250231,111250236,101250202,       111250202,111250228,111250235,111250243,111250252,       111250226,111250227,111250229,111250230,111250240)####定义一个查找函数，查找给定的学号是不是在我负责的名单里，####在就返回在元组里的编号，否则返回-1def find(num):    n=0    while n<len(tasks):        if(tasks[n]==num):            return n;        n = n+1    return -1files = os.listdir(name)###列出目录下所有文件curdir=os.path.join(os.getcwd(),name)###得到需要处理文件所在目录绝对路径print curdirfor f in files:    #print f    num = (int)(str(f)[:9])####前9位为学号    res = find(num)    fp = os.path.join(curdir,f)###当前处理的文件的绝对路径    #print fp    if res==-1: ##not finds        os.remove(fp) ##delete file删除文件，参数需要为绝对路径        print "remove "+f+"..."    else: ###rename to add team id         fn = "team_" + str((int)(res/5+37)) + "_" + f  ##文件名加所在组前缀        fnp= os.path.join(curdir,fn)        os.rename(fp,fnp)###重命名文件        print "rename "+fn#print filesprint "done."temp = input("press ENTER to exit...")

运行效果很好，比手工查找效率高的多。。。

脚本真好用。。。比c/c++/java之类的简单的多。。