reduce

pylibcudf.reduce.ScanType

See also cudf::scan_type.

Enum members

• INCLUSIVE

• EXCLUSIVE

pylibcudf.reduce.is_valid_reduce_aggregation(DataType source, Aggregation agg) → bool

Return if an aggregation is supported for a given datatype.

Parameters:

source

The type of the column the aggregation is being performed on.

agg

The aggregation.

Returns:

True if the aggregation is supported.

pylibcudf.reduce.minmax(Column col, Stream stream=None, DeviceMemoryResource mr=None) → tuple

Compute the minimum and maximum of a column

For details, see cudf::minmax documentation.

Parameters:

colColumn

The column to compute the minimum and maximum of.

streamStream | None

CUDA stream on which to perform the operation.

mrDeviceMemoryResource | None

Device memory resource used to allocate the returned scalars’ device memory.

Returns:

tuple

A tuple of two Scalars, the first being the minimum and the second being the maximum.

pylibcudf.reduce.reduce(Column col, Aggregation agg, DataType data_type, Scalar init=None, Stream stream=None, DeviceMemoryResource mr=None) → Scalar

Perform a reduction on a column

For details, see cudf::reduce documentation.

Parameters:

colColumn

The column to perform the reduction on.

aggAggregation

The aggregation to perform.

data_typeDataType

The data type of the result.

initScalar | None

The initial value for the reduction.

streamStream | None

CUDA stream on which to perform the operation.

mrDeviceMemoryResource | None

Device memory resource used to allocate the returned scalar’s device memory.

Returns:

Scalar

The result of the reduction.

pylibcudf.reduce.scan(Column col, Aggregation agg, scan_type inclusive, Stream stream=None, DeviceMemoryResource mr=None) → Column

Perform a scan on a column

For details, see cudf::scan documentation.

Parameters:

colColumn

The column to perform the scan on.

aggAggregation

The aggregation to perform.

inclusivescan_type

The type of scan to perform.

streamStream | None

CUDA stream on which to perform the operation.

mrDeviceMemoryResource | None

Device memory resource used to allocate the returned column’s device memory.

Returns:

Column

The result of the scan.