Skip to main content
This is a pre-production deployment of Warehouse. Changes made here affect the production instance of PyPI (pypi.python.org).
Help us improve Python packaging - Donate today!

Mutable, self-balancing interval tree

Project Description

NB: This package is deprecated. Please, use the intervaltree package instead (available via Github or PyPI).

The genome-related functionality is extracted to the intervaltree-bio package (Github, PyPI).

No future versions of this package are planned. Do not file issues.

A mutable, self-balancing interval tree. Queries may be by point, by range overlap, or by range envelopment.

This library was designed to allow tagging text and time intervals, where the intervals include the lower bound but not the upper bound.

Installation

The easiest way to install most Python packages is via easy_install or pip:

$ pip install PyIntervalTree

Features

  • Initialize blank or from an iterable of Intervals in O(n * log n).
  • Insertions
    • tree[begin:end] = data
    • tree.add(interval)
    • tree.addi(begin, end, data)
    • tree.extend(list_of_interval_objs)
  • Deletions
    • tree.remove(interval) (raises ValueError if not present)
    • tree.discard(interval) (quiet if not present)
    • tree.removei(begin, end, data)
    • tree.discardi(begin, end, data)
    • tree.remove_overlap(point)
    • tree.remove_overlap(begin, end) (removes all overlapping the range)
    • tree.remove_envelop(begin, end) (removes all enveloped in the range)
  • Overlap queries:
    • tree[point]
    • tree[begin:end]
    • tree.search(point)
    • tree.search(begin, end)
  • Envelop queries:
    • tree.search(begin, end, strict = True)
  • Membership queries:
    • interval_obj in tree (this is fastest, O(1))
    • tree.containsi(begin, end, data)
    • tree.overlaps(point)
    • tree.overlaps(begin, end)
  • Iterable:
    • for interval_obj in tree:
    • tree.items()
  • Sizing:
    • len(tree)
    • tree.is_empty()
    • not tree
    • tree.begin() (the smallest coordinate of the leftmost interval)
    • tree.end() (the end coordinate of the rightmost interval)
  • Restructuring
    • split_overlaps()
  • Copy- and typecast-able:
    • IntervalTree(tree) (Interval objects are same as those in tree)
    • tree.copy() (Interval objects are shallow copies of those in tree)
    • set(tree) (can later be fed into IntervalTree())
    • list(tree) (ditto)
  • Equal-able
  • Pickle-friendly
  • Automatic AVL balancing

Examples

  • Getting started:

    from intervaltree import Interval, IntervalTree
    t = IntervalTree()
    
  • Adding intervals - you don’t have to use strings!:

    t[1:2] = "1-2"
    t[4:7] = "4-7"
    t[5:9] = "5-9"
    
  • Query by point:

    ivs = t[6]            # set([Interval(4, 7, '4-7'), Interval(5, 9, '5-9')])
    iv = sorted(ivs)[0]   # Interval(4, 7, '4-7')
    
  • Accessing an Interval object:

    iv.begin  # 4
    iv.end    # 7
    iv.data   # "4-7"
    
  • Query by range:

    Note that ranges are inclusive of the lower limit, but non-inclusive of the upper limit. So:

    t[2:4]    # set([])
    

    But:

    t[1:5]    # set([Interval(1, 2, '1-2'), Interval(4, 7, '4-7')])
    
  • Constructing from lists of Interval’s:

    We could have made the same tree this way:

    ivs = [ [1,2], [4,7], [5,9] ]
    ivs = map( lambda begin,end: Interval(begin, end, "%d-%d" % (begin,end),
               *zip(*ivs) )
    
    t = IntervalTree(ivs)
    
  • Removing intervals:

    t.remove( Interval(1, 2, "1-2") )
    list(t)     # [Interval(4, 7, '4-7'), Interval(5, 9, '5-9')]
    
    t.remove( Interval(500, 1000, "Doesn't exist") # raises ValueError
    t.discard(Interval(500, 1000, "Doesn't exist") # quietly does nothing
    
    t.remove_overlap(5)
    list(t)     # []
    

    We could also empty a tree by removing all intervals, from the lowest bound to the highest bound of the IntervalTree:

    t.remove_overlap(t.begin(), t.end())
    

Usage with Genomic Data

Interval trees are especially commonly used in bioinformatics, where intervals correspond to genes or various features along the genome. Such intervals are commonly stored in BED-format files. To simplify working with such data, the package intervaltree.bio provides a GenomeIntervalTree class.

GenomeIntervalTree is essentially a dict of IntervalTree-s, indexed by chromosome names:

gtree = GenomeIntervalTree()
gtree['chr1'].addi(10000, 20000)

There is a convenience function for adding intervals:

gtree.addi('chr2', 20000, 30000)

You can create a GenomeIntervalTree instance from a BED file:

test_url = 'http://hgdownload.cse.ucsc.edu/goldenPath/hg19/encodeDCC/wgEncodeAwgTfbsUniform/wgEncodeAwgTfbsBroadDnd41Ezh239875UniPk.narrowPeak.gz'
data = zlib.decompress(urlopen(test_url).read(), 16+zlib.MAX_WBITS)
gtree = GenomeIntervalTree.from_bed(StringIO(data))

In addition, special functions are offered to read in UCSC tables of gene positions:

  • Load the UCSC knownGene table with each interval corresponding to gene’s transcribed region:

    knownGene = GenomeIntervalTree.from_table()
    
  • Load the UCSC refGene table with each interval corresponding to gene’s coding region:

    url = 'http://hgdownload.cse.ucsc.edu/goldenpath/hg19/database/refGene.txt.gz'
    refGene = GenomeIntervalTree.from_table(url=url, parser=UCSCTable.REF_GENE, mode='cds')
    
  • Load the UCSC ensGene table with each interval corresponding to a gene’s exon:

    url = 'http://hgdownload.cse.ucsc.edu/goldenpath/hg19/database/ensGene.txt.gz'
    ensGene = GenomeIntervalTree.from_table(url=url, parser=UCSCTable.ENS_GENE, mode='exons')
    

You may add methods for parsing your own tabular files with genomic intervals, see the documentation for GenomeIntervalTree.from_table.

Based on

Release History

Release History

This version
History Node

0.5

History Node

0.4

History Node

0.3

History Node

0.2.3

Download Files

Download Files

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

File Name & Checksum SHA256 Checksum Help Version File Type Upload Date
PyIntervalTree-0.5.zip (34.1 kB) Copy SHA256 Checksum SHA256 Source Jan 15, 2015

Supported By

WebFaction WebFaction Technical Writing Elastic Elastic Search Pingdom Pingdom Monitoring Dyn Dyn DNS Sentry Sentry Error Logging CloudAMQP CloudAMQP RabbitMQ Heroku Heroku PaaS Kabu Creative Kabu Creative UX & Design Fastly Fastly CDN DigiCert DigiCert EV Certificate Rackspace Rackspace Cloud Servers DreamHost DreamHost Log Hosting