Question

python 特定 行列 文本 统计 计数

文本1（file01)

要求：
1、统计文本1中第4列为A，第5列为G时出现的次数。
2、统计文本1中第4列为A，第5列为T时出现的次数。
3、统计文本1中第4列为A，第5列为C时出现的次数。
4、将统计好的次数按顺序写入新文本2（file02）中。

Answer 1

# open file
fin = open("file01.txt", "r")
fout = open("file02.txt", "w")
# init
count_AG = 0
count_AT = 0
count_AC = 0
# data lines
for line in fin:
    dat_in = line.split()
    if dat_in[3]=="A" and dat_in[4]=="G" :
        count_AG += 1
    if dat_in[3]=="A" and dat_in[4]=="T" :
        count_AT += 1
    if dat_in[3]=="A" and dat_in[4]=="C" :
        count_AC += 1        
# output
fout.write("count_AG: %s \n", count_AG)
fout.write("count_AT: %s \n", count_AT)
fout.write("count_AC: %s \n", count_AC)
# close file   
fin.close()
fout.close()

追问

谢谢你哈~不过最后#output部分报错，write()函数只能使用一个参数。
我改成fout.write("count_AG: %s \n" %(count_AG))。

追答

哈哈 是笔误了 谢谢指正~