From a8359efeada75a2bb9c3b70891baa57a2cc5c389 Mon Sep 17 00:00:00 2001
From: chaopower <chao.zhang@jmdna.com>
Date: Tue, 16 Jan 2024 10:32:25 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8C=96=E7=96=97bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 codes/chemo.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/codes/chemo.py b/codes/chemo.py
index 1283851..8bb388b 100755
--- a/codes/chemo.py
+++ b/codes/chemo.py
@@ -98,13 +98,7 @@ class ChemoRun:
             chrom = row['chr']
             end = row['end']
             fliter = pd.DataFrame()
-            if (chrom, end) not in records:
-                fliter = pd.concat([fliter, drug_rsid_data[
-                    (drug_rsid_data['chr'] == chrom) &
-                    (drug_rsid_data['end'] == end) &
-                    (drug_rsid_data['genotype'] == '0/0')
-                    ]])
-            else:
+            if (chrom, end) in records:
                 record = records[(chrom, end)]
                 ref = record.ref
                 alt = record.alts[0]
@@ -123,6 +117,14 @@ class ChemoRun:
                     (drug_rsid_data['alt'] == alt) &
                     (drug_rsid_data['genotype'] == gt)
                     ]])
+
+            if fliter.empty:
+                fliter = pd.concat([fliter, drug_rsid_data[
+                    (drug_rsid_data['chr'] == chrom) &
+                    (drug_rsid_data['end'] == end) &
+                    (drug_rsid_data['genotype'] == '0/0')
+                    ]])
+
             if fliter.empty:
                 raise UserWarning(
                     'chr: %s , end: %s 数据库未能匹配, 野生型0/0也未能匹配' % (chrom, end))