python - 使用txt文件中的数据回答python中的问题

Question

a01:01-24-2011:s1 
a03:01-24-2011:s2 
a02:01-24-2011:s2 
a03:02-02-2011:s2 
a03:03-02-2011:s1 
a02:04-19-2011:s2 
a01:05-14-2011:s2 
a02:06-11-2011:s2 
a03:07-12-2011:s1 
a01:08-19-2011:s1 
a03:09-19-2011:s1 
a03:10-19-2011:s2 
a03:11-19-2011:s1 
a03:12-19-2011:s2  

格式为animalid:date:location

以上是我保存在名为animallog2.txt的txt文件中的数据。我已将其导入到我的代码中。我想用上面的内容来回答几个问题。

到目前为止，这是我的代码

def main():
    fname = input("Enter name of file: ")
    with open(fname) as inf:
          animalnames, dates, locations = zip(*[line.strip().split(':') for line in inf])

d = {}
for animalname, loc in zip(animalnames, locations):
    d.setdefault(animalname, []).append(loc)  

print("Animal ID","Location1","Location2")       
for k, v in d.items():
    print(k, end='\t')                       #To print table of the data
    print(v.count('s1'), end='\t')
    print(v.count('s2'))        
print("================================================================================================================")

我的 table 是

Animal ID Location1 Location2
a01        2    1
a02        0    3

a03 4 4

有人可以帮助我如何使用以上所有内容来回答问题:

1) 访问两个站点至少 4 次的动物 #answer 是 a03

2)每只动物的总访问次数

Answer 1

您已经完成了大部分工作，这将得到您的答案:

1)

for k, v in d.items():
    if v.count('s1')>=4 and v.count('s2')>=4:
        print k

2)

for k, v in d.items():
    print k, v.count('s1')+v.count('s2')