擅长:python、mysql、java
<p>数据库是你的朋友我的儿子。</p>
<pre><code>import sqlite3
from Bio import SeqIO
db = sqlite3.connect("./db")
c = db.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS map (k text unique, v text)''')
db.commit()
def keys(db):
cursor = db.cursor()
return cursor.execute("""SELECT k FROM map""").fetchall()
def get(key, db, default=None):
cursor = db.cursor()
result = cursor.execute("""SELECT v FROM map WHERE k = ?""", (key,)).fetchone()
if result is None:
return default
return result[0]
def save(key, value, db):
cursor = db.cursor()
cursor.execute("""INSERT INTO map VALUES (?,?)""", (key, value))
db.commit()
with open('uniref90.fasta') as fasta_file:
for seq_record in SeqIO.parse(fasta_file, 'fasta'):
header = seq_record.id
uniID = header.split('_')[1]
seqs = str(seq_record.seq)
save(uniID, seqs, db)
</code></pre>