сохранение файла cvs в python - PullRequest
Новая
       16

сохранение файла cvs в python

0 голосов
/ 02 мая 2018

У меня проблемы с сохранением данных в CSV-файл. Я пытаюсь разделить аминокислоты в этом гене на четыре группы: неполярные (np), полярные (p), отрицательные (neg) и положительные (pos). Я выяснил, как подсчитать все отдельные аминокислоты и сохранить их в файле cvs, но не могу понять, как сохранить данные из четырех групп так же, как и с отдельными аминокислотами.

Вот код для сохранения одной аминокислоты в CSV: 

from Bio import Entrez, SeqIO

Entrez.email = ""

handle = Entrez.efetch(db="nucleotide", id="KT191142", rettype="gb", retmode="text")
record = SeqIO.read(handle, "genbank")
handle.close()

protein_seq = record.seq.translate()
print(protein_seq)

def count_aa(seq, csv_file):

aa_dict = {} # dictionary to store amino acid counts

for aa in seq:

    if aa in aa_dict:
        aa_dict[aa] += 1 # increment the count of an amino acid by 1
    else:
        aa_dict[aa] = 1 # set the count of an amino acid to 1

with open(csv_file, "w") as file:

    aa_list=sorted(aa_dict.keys())

    file.write("amino_acid,count\n")

    for aa in aa_list:
        line = str(aa) + ',' + str(aa_dict[aa]) + '\n'

        file.write(line)

count_aa(protein_seq, "ebola_aa_count2.csv")`

Я хочу сохранить этот новый код в CSV-файл, как и в предыдущем коде, вот новый код: 

import re

handle = Entrez.efetch(db="nucleotide", id="KT191142", rettype="gb", retmode="text")
records = SeqIO.read(handle, "genbank")
handle.close()

protein_seq = records.seq.translate()
print(protein_seq)

np_count = 0
p_count = 0
neg_count = 0
pos_count = 0

for aa in protein_seq:
    match_np = re.search(r"G|A|V|C|P|L|I|M|W|F", str(aa))
    match_p = re.search(r"S|T|Y|N|Q", str(aa))
    match_neg = re.search(r"D|E", str(aa))
    match_pos = re.search(r"K|R|H", str(aa))
    if match_np:
        np_count += 1
    if match_p:
        p_count += 1
    if match_neg:
        neg_count += 1
    if match_pos:
        pos_count += 1

handle.close()
print(np_count, p_count, neg_count, pos_count)

Заранее спасибо за помощь!

1 Ответ

0 голосов
/ 02 мая 2018

Вам определенно следует использовать панд для работы с CSV-файлами. 

import pandas as pd
df = pd.DataFrame(columns = ['np_count', 'p_count', 'neg_count', 'pos_count'], data =[np_count, p_count, neg_count, pos_count])
df.to_csv('output.csv')
...