今天在CU上无间看到一个帖子。

http://bbs.chinaunix.net/thread-3662056-1-1.html

需求如下

ci    pi       er          cc     22         ii
2     3        5           1       2         4
2     4        5           5       1         4
5     6        6           5       6         7
7     7        5           4       3         1

对文本的指定列求和求平均。

自己想一下，理了一下思路。发现和其他网友有一些不一样的东西。

 

 
  
  1. #/usr/bin/env python 
  2. ''' 
  3. USAGE: 
  4.         colsum.py filename colname 
  5. ''' 
  6. import sys 
  7.  
  8. def colSum(filename,colname): 
  9.         data=[] 
  10.         index=open(filename).readline().split().index(colname) 
  11.         for line in open(filename): 
  12.                 data.append(line.split()[index]) 
  13.         total=sum([int(i) for i in data[1:]]) 
  14.         avg=total/(len(data)-1) 
  15.         #return total,avg 
  16.         print "file '%s',col '%s':total is %s,avg is %s" % (filename,colname,total,avg) 
  17.  
  18. def main(): 
  19.         filename=sys.argv[1] 
  20.         colname=sys.argv[2] 
  21.         try: 
  22.                 f=open(filename) 
  23.         except IOError: 
  24.                 print "%s not found" % filename 
  25.         else: 
  26.                 colSum(filename,colname) 
  27.                 f.close() 
  28.  
  29. if __name__=='__main__': 
  30.         if len(sys.argv)==3: 
  31.                 main() 
  32.         else: 
  33.                 print __doc__ 
 
  
总结：
 
  
1、拆分，通过对象求索引
 
  
2、文件迭代
 
  
3、列表解析
 
  
看了其他网友使用for循环和while，考虑到性能问题和过于复杂使用了列表解析
 
  
4、思路
 
  
通过索引将矩阵中需要的数据抓取出来重组成列表求和，求平均。
5、其他工具
 
  
MS使用awk，更简单！！！