鱼C论坛

 找回密码
 立即注册
查看: 4355|回复: 1

python判断某个数值是否在区间范围内

[复制链接]
发表于 2018-11-30 15:20:11 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
有两个输入文件,一个是数值点文件probe.txt,共76行,格式如下:
chr1    13828906        13828907        rs7520386       0       +
chr1    88923260        88923261        rs7532151       0       +
chr1    107441475       107441476       rs776284        0       +

另一个文件input.bam,有很多行,每一行有一个start、end,现在就是想统计在input.bam里面分别有多少行能覆盖到probed.txt中的每一个点,最后的输出格式是:
chr1    13828906        13828907        rs7520386       0       +    12
chr1    88923260        88923261        rs7532151       0       +    8
chr1    107441475       107441476       rs776284        0       +    3

然后我的代码是:
  1. import pysam
  2. import pandas as pd

  3. start =time.clock()
  4. f1 = pysam.AlignmentFile("input.bam", 'rb')
  5. f3 = open("probe.txt",'r')
  6. df = pd.read_table("probe.txt",header=None)
  7. snp=[0] * 76
  8. flags = [99,83]

  9. def judge_range(line):
  10.     chr = line.reference_name
  11.     flag = line.flag
  12.     start = int(line.pos + 1)
  13.     end = int(start) + int(line.isize)
  14.     f3 = open('probe.txt','r')
  15.     col = 0
  16.     for i, region in enumerate(f3):
  17.         S = region.split("\t")[1]
  18.         CHR = region.split("\t")[0]
  19.         if chr == CHR and S in range(start,end):
  20.             col = i
  21.         else:
  22.             continue
  23.     return(col)
  24.          
  25. for line in f1:
  26.     flag = line.flag
  27.     if flag in flags:
  28.         col = judge_range(line)
  29.         snp[col] = snp[col] + 1
  30.     else:
  31.         continue
  32. data = pd.DataFrame(snp)
  33. data.rename(columns={0:'align_number'},inplace=True)
  34. df1 = pd.concat([df,data],axis=1,ignore_index=True)
  35. df1.to_csv('result.txt',header=False)
复制代码


请问一下这个代码该怎么修改才能提高运行效率,probe.txt文件只有76行,input.bam里面有上千万行,我现在写的这个一直在跑,就是跑不出结果。。。。。。。。。
小甲鱼最新课程 -> https://ilovefishc.com
回复

使用道具 举报

发表于 2018-11-30 21:46:16 | 显示全部楼层
请贴出文件probe.txt和input.bam
小甲鱼最新课程 -> https://ilovefishc.com
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2026-1-10 18:45

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表