统计的一个小题目python实现

最近面试碰到的一个题目，业余时间用python实现的。

拿到数据，先用sort 命令排序，也可再进一步去重复

sort -k 1,2  data.txt |uniq  > data.new

 # -*- coding:utf-8 -*-

 #第一个字段是用户uid,第二个字段是商品tid，统计每个商品的用户购买数量
 import sys

 pre_tid=""
 pre_uid=""
 sum=0
 for line in sys.stdin:
         fields = line.strip().split('\t') #注意这里要加上strip()去除"\n"
         uid = fields[0]
         tid = fields[1]
         if tid!=pre_tid:
                 if sum!=0: #读取第一行时，不要打印
                         print pre_tid+"\t"+str(sum) #类别改变就把上一个类别输出
                         pre_uid=""
                 pre_tid = tid
                 sum=1 #初始化计数器
         else:
                 if(uid==pre_uid): #如果有两条完全重复的数据不计入
                         continue
                 else:
                         sum=sum+1
                         pre_uid=uid

 print pre_tid+"\t"+str(sum)#打印最后一个类别