analyzer.core.columns

Attributes

logger
EVENTS

Classes

EventBackend	str(object='') -> str
Column	Simple wrapper for a "path" in events.
ColumnCollection
TrackedColumns

Functions

coerceFields(data)
setColumn(→ awkward.Array)	Set a potentially column in an array.
getAllColumns(→ set[Column])	Extract all columns from a potentially nested layout.
mergeColumns(column_views)
addSelection(columns, name, data)

Module Contents

analyzer.core.columns.logger

analyzer.core.columns.coerceFields(data)

analyzer.core.columns.EVENTS

class analyzer.core.columns.EventBackend

Bases: str, enum.Enum

str(object=’’) -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Create a new string object from the given object. If encoding or errors is specified, then the object must expose a data buffer that will be decoded using the given encoding and error handler. Otherwise, returns the result of object.__str__() (if defined) or repr(object). encoding defaults to sys.getdefaultencoding(). errors defaults to ‘strict’.

coffea_virtual = 'coffea_virtual'

coffea_dask = 'coffea_dask'

coffea_imm = 'coffea_eager'

rdf = 'rdf'

class analyzer.core.columns.Column

Simple wrapper for a “path” in events. Stores a single tuple representing the path to the desired field. Automatically converts “dot” delimited paths.

fields: tuple[str, Ellipsis]

contains(other: Column | str | tuple[str, Ellipsis])

extract(events: awkward.Array)

__iter__()

__eq__(other)

__len__()

__getitem__(key: int)

__add__(other: Column)

__radd__(other)

__hash__()

__str__()

__repr__()

property adl_name

parents()

analyzer.core.columns.setColumn(events: awkward.Array, column: Column, value: awkward.Array) → awkward.Array

Set a potentially column in an array. Automatically creates intermediate fields if needed.

class analyzer.core.columns.ColumnCollection

columns: set[Column]

__iter__()

contains(other: Column)

Check if any of the columns in the set is a parent of other.

intersect(other: ColumnCollection)

analyzer.core.columns.getAllColumns(events, cur_col=None, cur_depth=0, max_depth=None) → set[Column]

Extract all columns from a potentially nested layout.

class analyzer.core.columns.TrackedColumns

INTERNAL_USE_COL: ClassVar[Column]

backend: EventBackend

metadata: Any | None = None

pipeline_data: dict[str, Any]

property events

flush()

Synchronize stored lazy columns with underlying events

property fields

updatedColumns(old, limit=None)

Determine which in the current TrackedColumns have been updated from old by comparing their provenance.

copy()

static fromEvents(events, metadata, backend, provenance: int)

getKeyForColumns(columns)

Get an excecution key for the column. Returns a hash dependent on the provenance of all the columns contains in the input.

getKeyForAll()

__setitem__(column, value)

Set a column, potentially lazy.

__getitem__(column)

Get a column. May use the lazy cache.

addColumnsFrom(other, columns)

Merge in columns from another TrackedColumns.

filter(mask)

Filter at the event level.

useKey(provenance)

allowedInputs(columns: list[Column] | ColumnCollection)

allowedOutputs(columns: list[Column] | ColumnCollection)

analyzer.core.columns.mergeColumns(column_views)

analyzer.core.columns.addSelection(columns, name, data)