Add files via upload

ishworthapa · Aug 24, 2019 · 3252e39 · 3252e39
1 parent a89d7d1
commit 3252e39
Showing 1 changed file with 45 additions and 0 deletions.
diff --git a/src/ref_sel_test_symbol.py b/src/ref_sel_test_symbol.py
@@ -0,0 +1,45 @@
+import codecs
+import csv
+import pandas
+import sys, operator
+
+
+def getRefExpLevel(lowerTh, upperTh, outputfile):
+    with codecs.open("final_out.csv", "r", encoding='utf-8', errors='ignore') as f:
+        with open(outputfile, 'w') as csvfile:
+          reader = csv.DictReader(f)
+          writer = csv.writer(csvfile)
+          writer.writerow(["genes", "cov", "mean", "std", "MFC", "covMFC"])
+          for row in reader:
+            #try:
+
+                ens = []
+                if lowerTh <= pandas.to_numeric(row["mean"]) <= upperTh:
+                  covMFC = float(row["MFC"])*float(row["cov"])
+                  Account = [row["genes"], row["cov"], row["mean"], row["std"], row["MFC"], covMFC]
+                #print(row["Gene_ID"] + " " + mg.getgene(row["Gene_ID"])['symbol'])
+                  writer.writerow(Account)
+
+def findTop(resultFile):
+    df  = pandas.read_csv(resultFile)
+    sortedlist = df.sort_values(by='covMFC')
+    for index, row in sortedlist.iterrows():
+        if row[0] in lines:
+            print(row[0])
+            break
+
+
+text_file = open("data/housekeeping_symbol.txt", "r")
+lines = text_file.read().splitlines()
+
+outfiles = ['ref1.csv','ref2.csv','ref3.csv','ref4.csv']
+lows = [3.75, 5, 6.5, 9]
+ups = [4.25, 5.5, 7, 9.5]
+
+for i in range(len(outfiles)):
+    getRefExpLevel(lows[i],ups[i],outfiles[i])
+
+for outfile in outfiles:
+    findTop(outfile)
+
+sys.exit(0)