ListSlice

class nvtabular.ops.ListSlice(start, end=None)[source]

Bases: nvtabular.ops.operator.Operator

Slices a list column

This operator provides the ability to slice list column by row. For example, to truncate a list column to only include the first 10 elements per row:

truncated = column_names >> ops.ListSlice(10)

Take the first 10 items, ignoring the first element:

truncated = column_names >> ops.ListSlice(1, 11)

Take the last 10 items from each row:

truncated = column_names >> ops.ListSlice(-10)
transform(col_selector: nvtabular.columns.selector.ColumnSelector, df: pandas.core.frame.DataFrame) → pandas.core.frame.DataFrame[source]

Transform the dataframe by applying this operator to the set of input columns

Parameters

  • columns (list of str or list of list of str) – The columns to apply this operator to

  • df (Dataframe) – A pandas or cudf dataframe that this operator will work on

Returns

Returns a transformed dataframe for this operator

Return type

DataFrame

output_tags()[source]