biotite.structure.AtomArrayStack

class biotite.structure.AtomArrayStack(depth, length)[source]

Bases: biotite.structure.atoms._AtomArrayBase

A collection of multiple AtomArray instances, where each atom array has equal annotation arrays.

Effectively, this means that each atom is occuring in every array in the stack at differing coordinates. This situation arises e.g. in NMR-elucidated or simulated structures. Since the annotations are equal for each array the annotaion arrays are 1-D, while the coordinate array is 3-D (m x n x 3).

Indexing works similar to AtomArray, with the difference, that two index dimensions are possible: The first index dimension specifies the array(s), the second index dimension specifies the atoms in each array (same as the index in AtomArray). Using a single integer as first dimension index returns a single AtomArray instance.

Concatenation of atoms for each array in the stack is done using the ‘+’ operator. For addition of atom arrays onto the stack use the stack() method.

The box attribute has the shape m x 3 x 3, as the cell might be different for each frame in the atom array stack.

Parameters
depthint

The fixed amount of arrays in the stack. When indexing, this is the length of the first dimension.

lengthint

The fixed amount of atoms in each array in the stack. When indexing, this is the length of the second dimension.

See also

AtomArray

Examples

Creating an atom array stack from two arrays:

>>> atom1 = Atom([1,2,3], chain_id="A")
>>> atom2 = Atom([2,3,4], chain_id="A")
>>> atom3 = Atom([3,4,5], chain_id="B")
>>> atom_array1 = array([atom1, atom2, atom3])
>>> print(atom_array1.coord)
[[1. 2. 3.]
 [2. 3. 4.]
 [3. 4. 5.]]
>>> atom_array2 = atom_array1.copy()
>>> atom_array2.coord += 3
>>> print(atom_array2.coord)
[[4. 5. 6.]
 [5. 6. 7.]
 [6. 7. 8.]]
>>> array_stack = stack([atom_array1, atom_array2])
>>> print(array_stack.coord)
[[[1. 2. 3.]
  [2. 3. 4.]
  [3. 4. 5.]]
<BLANKLINE>
 [[4. 5. 6.]
  [5. 6. 7.]
  [6. 7. 8.]]]
Attributes
{annot}ndarray, shape=(n,)

Mutliple n-length annotation arrays.

coordndarray, dtype=float, shape=(m,n,3)

ndarray containing the x, y and z coordinate of the atoms.

bonds: BondList or None

A BondList, specifying the indices of atoms that form a chemical bond.

box: ndarray, dtype=float, shape=(m,3,3) or None

The surrounding box. May represent a MD simulation box or a crystallographic unit cell.

get_array(self, index)[source]

Obtain the atom array instance of the stack at the specified index.

The same as stack[index], if index is an integer.

Parameters
indexint

Index of the atom array.

Returns
arrayAtomArray

AtomArray at position index.

stack_depth(self)[source]

Get the depth of the stack.

This value represents the amount of atom arrays in the stack. It is the same as len(array).

Returns
lengthint

Length of the array(s).

add_annotation(self, category, dtype)

Add an annotation category, if not already existing.

Initially the new annotation is filled with the zero representation of the given type.

Parameters
categorystr

The annotation category to be added.

dtypetype or str

A type instance or a valid NumPy dtype string. Defines the type of the annotation

See also

set_annotation
array_length(self)

Get the length of the atom array.

This value is equivalent to the length of each annotation array. For AtomArray it is the same as len(array).

Returns
lengthint

Length of the array(s).

copy(self)

Create a deep copy of this object.

Returns
copy

A copy of this object.

del_annotation(self, category)

Removes an annotation category.

Parameters
categorystr

The annotation category to be removed.

equal_annotation_categories(self, item)

Check, if this object shares equal annotation array catgeories with the given AtomArray or AtomArrayStack.

Parameters
itemAtomArray or AtomArrayStack

The object to compare the annotation arrays with.

Returns
equalitybool

True, if the annotation array names are equal.

equal_annotations(self, item)

Check, if this object shares equal annotation arrays with the given AtomArray or AtomArrayStack.

Parameters
itemAtomArray or AtomArrayStack

The object to compare the annotation arrays with.

Returns
equalitybool

True, if the annotation arrays are equal.

get_annotation(self, category)

Return an annotation array.

Parameters
categorystr

The annotation category to be returned.

Returns
arrayndarray

The annotation array.

get_annotation_categories(self)

Return a list containing all annotation array categories.

Returns
categorieslist

The list containing the names of each annotation array.

set_annotation(self, category, array)

Set an annotation array. If the annotation category does not exist yet, the category is created.

Parameters
categorystr

The annotation category to be set.

arrayndarray or None

The new value of the annotation category. The size of the array must be the same as the array length.