Update Sort_Scores.py

Sort_Scores.py

@@ -1,19 +1,19 @@
-import streamlit as st
+import re
 from tname import *
+from Rfile import *
 
 
 def Sort_Scores(file):
     scores = []
-    # file_path = st.text_input('请输入文件路径：', )
-    # save_path = st.text_input('请输入文件保存路径：', )
-    # if st.button('文件路径输入完成'):
-    with open(file.name, "r") as f:
-        contents = f.readlines()
+    contents = j_reads(file.name)
 
     for i in range(2, len(contents)):
         if i % 2 == 0:
-            content = contents[i].split()
-            scores.append(float(content[2][6:-1]))  # 存储scores值
+            # 使用正则表达式
+            content = re.match('.*score=(\d.\d+?),', contents[i])
+            if content:
+                score = content.group(1)
+                scores.append(float(score))
 
     na = Name()
     na = na + r"scores_sort.tsv"  # 结果文件名称
@@ -24,23 +24,12 @@ def Sort_Scores(file):
 
     # 第一条序列和其他序列格式不一样，且第一条序列不需要排序，单独写入
     with open(na, "a") as f1:
-        # f1.write(contents[0])
-        content = contents[0].split()
-        f1.write(content[0])
-        f1.write("\t")
-        f1.write(content[1])
-        f1.write("\n")
+        f1.write(contents[0])
         f1.write(contents[1])
 
     for i in range(0, len(scores)):
-        with open(na, "a") as f1:
-            content = contents[sorted_id[i] * 2 + 2].split()
-            f1.write(content[0])
-            f1.write("\t")
-            f1.write(content[2])
-            f1.write("\t")
-            f1.write(content[3])
-            f1.write("\n")
-            # f1.write(contents[sorted_id[i] * 2 + 2])  #由于文件前两行未参与排序，所以索引要+2
-            f1.write(contents[sorted_id[i] * 2 + 2 + 1])
+        with open(na, "a") as f:
+            f.write(contents[sorted_id[i] * 2 + 2])
+            # 由于文件前两行未参与排序，所以索引要+2
+            f.write(contents[sorted_id[i] * 2 + 2 + 1])
     return na