Skip to main content

bitarray: module for efficiently storing bits in a list-like object

Project description

This module provides an object type which efficiently represents an array of booleans. Bitarrays are sequence types and behave very much like usual lists. Each bit is represented as an actual bit in memory in a single array of bytes. Most of the functionality is implemented in C. Methods for converting accessing the machine representation are provided. This can be useful when bit level access to binary files is required, such as portable bitmap image files (.pbm). Also, when dealing with compressed date which uses variable bit length encoding, you will find this module useful.

Requires Python 2.5 or greater, see PEP 353.

Installation

bitarray can be installed from source:

$ tar xzf bitarray-0.2.2.tar.gz
$ cd bitarray-0.2.2
$ python setup.py install

On Unix systems, the latter command may have to be executed with root privileges. If you have setuptools installed, you can easy_install bitarray. Once you have installed the package, you may want to test it:

$ python -c 'import bitarray; bitarray.test()'
bitarray is installed in: /usr/local/lib/python2.5/site-packages/bitarray
bitarray version: 0.2.2
2.5.2 (r252:60911, Jul 17 2008, 10:38:24)
[GCC 4.2.1 (SUSE Linux)]
..................................................................
----------------------------------------------------------------------
Ran 66 tests in 0.454s

OK

You can always import the function test, and test().wasSuccessful() will return True when the test went OK.

Using the module

As mentioned above, bitarray objects behave very much like lists, so there is not too new to learn. The biggest difference to list objects is the ability to access the machine representation of the object. When doing so, the bit endianness is of importance, this issue is explained in detail in the section below. Here, we demonstrate the basic usage of bitarray objects:

>>> from bitarray import bitarray
>>> a = bitarray()            # create empty bitarray
>>> a.append(True)
>>> a.extend([False, True, True])
>>> a
bitarray('1011')

Bitarray objects can be instantiated in different ways:

>>> a = bitarray(2**20)       # bitarray of length 1048576 (uninitialized)
>>> bitarray('1001011')       # from a string
bitarray('1001011')
>>> lst = [True, False, False, True, False, True, True]
>>> bitarray(lst)             # from list, tuple, iterable
bitarray('1001011')

Bits can be assigned from any Python object, if the value can be interpreted as a truth value. You can think of this as Python’s built-in function bool() being applied, whenever casting an object:

>>> a = bitarray([42, '', True, {}, 'foo', None])
>>> a
bitarray('101010')
>>> a.append(a)      # note that bool(a) is True
>>> a.count(42)      # counts occurrences of True (not 42)
4L
>>> a.remove('')     # removes first occurrence of False
>>> a
bitarray('110101')

Like lists, bitarray objects support slice assignment and deletion:

>>> a = bitarray(50)
>>> a.setall(False)
>>> a[11:37:3] = 9 * bitarray([True])
>>> a
bitarray('00000000000100100100100100100100100100000000000000')
>>> del a[12::3]
>>> a
bitarray('0000000000010101010101010101000000000')
>>> a[-6:] = bitarray('10011')
>>> a
bitarray('000000000001010101010101010100010011')
>>> a += bitarray('000111')
>>> a[9:]
bitarray('001010101010101010100010011000111')

Bit endianness

Since a bitarray allows addressing of individual bits, where the machine represents 8 bits in one byte, there two obvious choices for this mapping; little- and big-endian. When creating a new bitarray object, the endianness can always be specified explicitly:

>>> a = bitarray(endian='little')
>>> a.fromstring('A')
>>> a
bitarray('10000010')
>>> b = bitarray('11000010', endian='little')
>>> b.tostring()
'C'

Here the low-bit comes first because little-endian means that increasing numeric significance corresponds to an increasing address (or index). So a[0] is the lowest and least significant bit, and a[7] is the highest and most significant bit.

>>> a = bitarray(endian='big')
>>> a.fromstring('A')
>>> a
bitarray('01000001')
>>> a[6] = 1
>>> a.tostring()
'C'

Here the high-bit comes first because big-endian means “most-significant first”. So a[0] is now the lowest and most significant bit, and a[7] is the highest and least significant bit.

The bit endianness is a property attached to each bitarray object. When comparing bitarray objects, the endianness (and hence the machine representation) is irrelevant; what matters is the mapping from indices to bits:

>>> bitarray('11001', endian='big') == bitarray('11001', endian='little')
True

When converting to and from machine representation, using the tostring, fromstring, tofile and fromfile methods, the endianness matters:

>>> a = bitarray(endian='little')
>>> a.fromstring('\x01')
>>> a
bitarray('10000000')
>>> b = bitarray(endian='big')
>>> b.fromstring('\x80')
>>> b
bitarray('10000000')
>>> a == b
True
>>> a.tostring() == b.tostring()
False

The endianness can not be changed once an object is created. However, since creating a bitarray from another bitarray just copies the memory representing the data, you can create a new bitarray with different endianness:

>>> a = bitarray('11100000', endian='little')
>>> a
bitarray('11100000')
>>> b = bitarray(a, endian='big')
>>> b
bitarray('00000111')
>>> a == b
False
>>> a.tostring() == b.tostring()
True

The default bit endianness is currently big-endian, however this may change in the future, and when dealing with the machine representation of bitarray objects, it is recommended to always explicitly specify the endianness.

Unless, explicity converting to machine representation, using the tostring, fromstring, tofile and fromfile methods, the bit endianness will have no effect on any computation, and you can safely ignore setting the endianness, and other details of this section.

Reference

The bitarray class:

bitarray([initial][endian=string])

Return a new bitarray object whose items are bits initialized from the optional initial, and endianness. If no object is provided, the bitarray is initialized to have length zero. The initial object may be of the following types:

int, long

Create bitarray of length given by the integer. The initial values in the array are random, because only the memory allocated.

string

Create bitarray from a string with ‘0’s and ‘1’s.

list, tuple, iterable

Create bitarray from a sequence, each element in the sequence is converted to a bit using truth value value.

bitarray

Create bitarray from another bitarray. This is done by copying the memory holding the bitarray data, and is hence very fast.

The optional keyword arguments ‘endian’ specifies the bit endianness of the created bitarray object. Allowed values are ‘big’ and ‘little’ (default is ‘big’).

Note that setting the bit endianness only has an effect when accessing the machine representation of the bitarray, i.e. when using the methods: tofile, fromfile, tostring, fromstring

A bitarray object supports the following methods:

all()

Returns True when all bits in the array are True.

any()

Returns True when any bit in the array is True.

append(x)

Append the value bool(x) to the end of the bitarray.

buffer_info()

Return a tuple (address, size, endianness, unused) giving the current memory address, the size (in bytes) used to hold the bitarray’s contents, the bit endianness as a string, and the number of unused bits (0..7). For example, a bitarray of length 11 will have a buffer size of 2 bytes and 5 unused bits.

bytereverse()

For all bytes representing the bitarray, reverse the bit order (in place). Note: This method changes the actual machine values representing the bitarray; it does not change the endianness of the bitarray object.

count(x)

Return number of occurrences of x in the bitarray.

endian()

Return the bit endianness as a string (either ‘little’ or ‘big’).

extend(object)

Append bits to the end of the bitarray. The objects which can be passed to this method are the same which can given to a bitarray object upon upon initialization.

fill()

Returns the number of bits added (0..7) at the end of the array. When the length of the bitarray is not a multiple of 8, increase the length slightly such that the new length is a multiple of 8, and set the few new bits to False.

from01(string)

Appends items from a string, containing ‘0’s and ‘1’s, to the bitarray. This method is deprecated, use the extend method instead.

fromfile(f [, n])

Read n bytes from the file object f and append them to the bitarray interpreted as machine values. When n is omitted, as many bytes are read until EOF is reached.

fromlist(list)

Append bits to bitarray object from an ordinary list. This method is deprecated, use the extend method instead.

fromstring(string)

Append from a string, interpreting the string as machine values.

index(x)

Return index of first occurrence of x in the bitarray. It is an error when x does not occur in the bitarray

insert(i, x)

Insert a new item x into the bitarray before position i.

invert(x)

Invert all bits in the array (in place), i.e. convert each 1-bit into a 0-bit and vice versa.

length()

Return the length (number of bits) of the bitarray.

pop([i])

Return the i-th element and delete it from the bitarray. i defaults to -1.

remove(x)

Remove the first occurrence of x in the bitarray.

reverse()

Reverse the order of bits in the array (in place).

setall(x)

Set all bits in the bitarray to bool(x).

sort()

Sort the bits in the array (in place).

to01())

Return a string containing ‘0’s and ‘1’s, representing the bits in the bitarray object.

tofile(f)

Write all bits (as machine values) to the file object f. When the length of the bitarray is not a multiple of 8, the remaining bits (1..7) are set to 0.

tolist()

Return an ordinary list with the items in the bitarray.

tostring()

Return the string representing (machine values) of the bitarray. When the length of the bitarray is not a multiple of 8, the few remaining bits (1..7) are set to 0.

Functions defined in the module:

test(verbosity=1)

Run a unittest for the module.

bits2bytes(n)

Return the number of bytes necessary to store n bits.

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

bitarray-0.2.2.tar.gz (34.0 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page