랜덤하게 단백질 서열만들기

# python 2.7 버전
# -*- coding: utf-8 -*-
from random import Random

def AminoacidGenerator(number, length):
    print '***generate random aminoacid sequence ***'
    codeFile = open('codes.txt', 'w')
    if number <= 0:
        return 'invalid number of protein'
    else:
        aminoacid = 'ACDEFGHIKLMNPQRSTVWY'      
        random = Random()
        for j in range(1, number + 1):
            str = ''
            for i in range(1, length+1):
                index = random.randint(1,len(aminoacid))
                str = str + aminoacid[index -1]
            codeFile.write(str+'\n')

# 100개 길이의 단백질 서열을 100개 만들어 봅니다. 
print AminoacidGenerator(100, 100)