biotite.structure.get_chain_masks

biotite.structure.get_chain_masks(array, indices)[source]

Get boolean masks indicating the chains to which the given atom indices belong.

Parameters
arrayAtomArray, shape=(n,) or AtomArrayStack, shape=(m,n)

The atom array (stack) to determine the chains from.

indicesndarray, dtype=int, shape=(k,)

These indices indicate the atoms to get the corresponding chains for. Negative indices are not allowed.

Returns
chains_masksndarray, dtype=bool, shape=(k,n)

Multiple boolean masks, one for each given index in indices. Each array masks the atoms that belong to the same chain as the atom at the given index.