ai.rapids.cudf

Class Table

java.lang.Object

ai.rapids.cudf.Table

All Implemented Interfaces:

AutoCloseable

public final class Table
extends Object
implements AutoCloseable

Class to represent a collection of ColumnVectors and operations that can be performed on them collectively. The refcount on the columns will be increased once they are passed in

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Table.AggregateOperation Class representing aggregate operations
static class	Table.OrderByArg
static class	Table.TableOperation
static class	Table.TestBuilder Create a table on the GPU with data from the CPU.

Constructor Summary

Constructors

Constructor and Description
Table(ColumnVector... columns) Table class makes a copy of the array of ColumnVectors passed to it.

Method Summary

Modifier and Type	Method and Description
static Table.OrderByArg	asc(int index)
static Aggregate	avg(int index)
void	close()
static Table	concatenate(Table... tables) Concatenate multiple tables together to form a single table.
static Aggregate	count()
static Table.OrderByArg	desc(int index)
ColumnVector	getColumn(int index) Return the ColumnVector at the specified index.
int	getNumberOfColumns()
long	getRowCount()
Table.AggregateOperation	groupBy(int... indices)
static Aggregate	max(int index)
static Aggregate	min(int index)
Table.TableOperation	onColumns(int... indices)
Table	orderBy(boolean areNullsSmallest, Table.OrderByArg... args) Orders the table using the sortkeys returning a new allocated table.
static Table	readCSV(Schema schema, byte[] buffer)
static Table	readCSV(Schema schema, CSVOptions options, byte[] buffer)
static Table	readCSV(Schema schema, CSVOptions opts, byte[] buffer, long offset, long len)
static Table	readCSV(Schema schema, CSVOptions opts, File path)
static Table	readCSV(Schema schema, CSVOptions opts, HostMemoryBuffer buffer, long len)
static Table	readCSV(Schema schema, File path)
static Table	readParquet(byte[] buffer)
static Table	readParquet(File path)
static Table	readParquet(ParquetOptions opts, byte[] buffer)
static Table	readParquet(ParquetOptions opts, byte[] buffer, long len)
static Table	readParquet(ParquetOptions opts, File path)
static Table	readParquet(ParquetOptions opts, HostMemoryBuffer buffer, long len)
static Aggregate	sum(int index)
String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

Table

public Table(ColumnVector... columns)

Table class makes a copy of the array of ColumnVectors passed to it. The class will decrease the refcount on itself and all its contents when closed and free resources if refcount is zero

Parameters:

columns - - Array of ColumnVectors

Method Detail

getColumn

public ColumnVector getColumn(int index)

Return the ColumnVector at the specified index. If you want to keep a reference to the column around past the life time of the table, you will need to increment the reference count on the column yourself.

getRowCount

public final long getRowCount()

getNumberOfColumns

public final int getNumberOfColumns()

close

public void close()

Specified by:

close in interface AutoCloseable

toString

public String toString()

Overrides:

toString in class Object

readCSV

public static Table readCSV(Schema schema,
                            File path)

readCSV

public static Table readCSV(Schema schema,
                            CSVOptions opts,
                            File path)

readCSV

public static Table readCSV(Schema schema,
                            byte[] buffer)

readCSV

public static Table readCSV(Schema schema,
                            CSVOptions options,
                            byte[] buffer)

readCSV

public static Table readCSV(Schema schema,
                            CSVOptions opts,
                            byte[] buffer,
                            long offset,
                            long len)

readCSV

public static Table readCSV(Schema schema,
                            CSVOptions opts,
                            HostMemoryBuffer buffer,
                            long len)

readParquet

public static Table readParquet(File path)

readParquet

public static Table readParquet(ParquetOptions opts,
                                File path)

readParquet

public static Table readParquet(byte[] buffer)

readParquet

public static Table readParquet(ParquetOptions opts,
                                byte[] buffer)

readParquet

public static Table readParquet(ParquetOptions opts,
                                byte[] buffer,
                                long len)

readParquet

public static Table readParquet(ParquetOptions opts,
                                HostMemoryBuffer buffer,
                                long len)

concatenate

public static Table concatenate(Table... tables)

Concatenate multiple tables together to form a single table. The schema of each table (i.e.: number of columns and types of each column) must be equal across all tables and will determine the schema of the resulting table.

orderBy

public Table orderBy(boolean areNullsSmallest,
                     Table.OrderByArg... args)

Orders the table using the sortkeys returning a new allocated table. The caller is responsible for cleaning up the ColumnVector returned as part of the output Table

Example usage: orderBy(true, Table.asc(0), Table.desc(3)...);

Parameters:

areNullsSmallest - - represents if nulls are to be considered smaller than non-nulls.

args - - Suppliers to initialize sortKeys.

Returns:

Sorted Table

asc

public static Table.OrderByArg asc(int index)

desc

public static Table.OrderByArg desc(int index)

count

public static Aggregate count()

max

public static Aggregate max(int index)

min

public static Aggregate min(int index)

sum

public static Aggregate sum(int index)

avg

public static Aggregate avg(int index)

groupBy

public Table.AggregateOperation groupBy(int... indices)

onColumns

public Table.TableOperation onColumns(int... indices)