-
Notifications
You must be signed in to change notification settings - Fork 1
/
output_to_csv.py
72 lines (52 loc) · 2.43 KB
/
output_to_csv.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import sys
import pandas as pd
'''
####################################################################################################
To use this script:
python output_to_csv.py [INPUT TYPE] [SEQUENCE NAME] [INPUT VALUES]
Example:
python output_to_csv.py e6 1aa7 0.10000094301032349,0.8701883352082787,0.053361433898144774,
0.053092652126686124,0.1940914724635971,0.9483219853442273,0.3306576724498245,0.6879327099561908,
1.2548325790728785,0.07668346215539201,0.1396246476315025,0.27523407620439627,0.012628703101327032,
0.10997895755054084,0.03559891746031243,0.2870047965449099,0.11392988629966698,0.20413947585217165,
0.14582306776830062,0.11202065669472744,0.37565077552345605
Note:
The output of the above example is a file called "e6_1aa7.csv" located in the current working
directory.
If there are problems with this script please contact: [email protected]
#####################################################################################################
'''
if len(sys.argv) == 4:
sequence_name = sys.argv[2]
input_length = len(sys.argv[3])
values_input = sys.argv[3][0:input_length]
values_input = values_input.split(',')
input_type = sys.argv[1]
else:
print('')
print('Remember, next time you can enter arguments directly into the command line terminal.')
print('')
print('To do this enter the following command: ')
print('python output_to_csv.py [E6, E20, or VKBAT] [PROTEIN ID] [INPUT VALUES]')
print('')
print('For now, just enter the information in the following prompts.')
print('')
input_type = input('PLEASE ENTER THE INPUT TYPE (ie. e6, e20, or vkbat): ')
sequence_name = input('PLEASE ENTER THE PROTEIN NAME (ie. 1AA7): ')
values = input('PLEASE PASTE THE RAW E6, E20, OR VKBAT VALUES: ')
input_length = len(values)
values_input = values[0:input_length]
values_input = values_input.split(',')
# make output list variable
output = []
# loop through the input and append to list
for number in values_input:
output.append(number)
# generate dataframe
df = pd.DataFrame()
df[str(input_type)] = output
# prints dataframe to terminal (testing purposes)
#print(df)
print('Successfully created ' + str(input_type) + '_' + str(sequence_name) + '.csv' + ' in the current working directory!')
# generate the csv with the output values
df.to_csv(str(input_type)+'_'+str(sequence_name)+'.csv')