DataGenerator.py 文件源码

def getFeaturesForGenome(genomeId, CDS_ONLY):
    """
    This method gets the features for a particular genomeId frfom PATRIC

    Parameters

    genomeId: UniqueId for the genome
    CDS_ONLY: retrieve only CDS features
    """
    data_table = pd.read_table(PatricURL
                               +genomeId+'/'+genomeId+'.PATRIC.features.tab')


    print data_table.shape

    if CDS_ONLY:
        return data_table[(data_table.feature_type == 'CDS')]

    else:
        return data_table