Skip to main content

GO enrichment with python -- pandas meets networkx

Project description


Convenient GO enrichments from python. For use in python projects.

  1. Builds the GO-ontology graph
  2. Propagates GO-annotations up the graph
  3. Performs enrichment test for all categories
  4. Performs multiple testing correction
  5. Allows for export to pandas for processing and graphviz for visualization


Install package from pypi and download ontology and needed annotations.
pip install goenrich
mkdir db
# Ontology
wget -O db/go-basic.obo
# UniprotACC
wget -O db/gene_association.goa_human.gaf.gz
# Yeast SGD
wget -O db/gene_association.sgd.gz
# Entrez GeneID
wget -O db/gene2go.gz

Run GO enrichment

import goenrich

# build the ontology
O = goenrich.obo.ontology('db/go-basic.obo')

# use all entrez geneid associations form gene2go as background
# use annot ='db/gene_association.goa_human.gaf.gz') for uniprot
# use annot ='db/gene_association.sgd.gz') for yeast
gene2go ='db/gene2go.gz')
# use values = {k: set(v) for k,v in annot.groupby('go_id')['db_object_symbol']} for uniprot/yeast
values = {k: set(v) for k,v in gene2go.groupby('GO_ID')['GeneID']}

# propagate the background through the ontology
background_attribute = 'gene2go'
goenrich.enrich.propagate(O, values, background_attribute)

# extract some list of entries as example query
# use query = annot['db_object_symbol'].unique()[:20]
query = gene2go['GeneID'].unique()[:20]

# for additional export to graphviz just specify the gvfile argument
# the show argument keeps the graph reasonably small
df = goenrich.enrich.analyze(O, query, background_attribute, gvfile='')

# generate html

# call to graphviz
import subprocess
subprocess.check_call(['dot', '-Tpng', '', '-o', 'test.png'])

Generate png image using graphviz:

dot -Tpng > example.png

or directly from python:

import subprocess
subprocess.check_call(['dot', '-Tpng', '', '-o', 'example.png'])

Check the documentation for all available parameters

Licence & Contributors

This work is licenced under the MIT licence

Contributions are welcome!

Special thanks

  • @lukauskas for implementing i/o support for file-like objects.
  • @zfrenchee for fixing a bug in the calculation of the test statistic.
  • @pommy1 for implementing support for networkx >= 2.0.0.

Building the documentation

sphinx-apidoc -f -o docs goenrich goenrich/tests

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for goenrich, version 1.12
Filename, size File type Python version Upload date Hashes
Filename, size goenrich-1.12.tar.gz (28.1 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page