UNPKG

apache-arrow

Version:
40 lines (39 loc) 2.06 kB
import { Table } from '../table'; import { Vector } from '../vector'; import { Predicate, Col } from './predicate'; import { RecordBatch } from '../recordbatch'; import { VectorType as V } from '../interfaces'; import { DataType, Int, Struct } from '../type'; /** @ignore */ export declare type BindFunc = (batch: RecordBatch) => void; /** @ignore */ export declare type NextFunc = (idx: number, batch: RecordBatch) => void; export declare class DataFrame<T extends { [key: string]: DataType; } = any> extends Table<T> { filter(predicate: Predicate): FilteredDataFrame<T>; scan(next: NextFunc, bind?: BindFunc): void; scanReverse(next: NextFunc, bind?: BindFunc): void; countBy(name: Col | string): CountByResult<any, Int<import("../enum").Type.Int | import("../enum").Type.Int8 | import("../enum").Type.Int16 | import("../enum").Type.Int32 | import("../enum").Type.Int64 | import("../enum").Type.Uint8 | import("../enum").Type.Uint16 | import("../enum").Type.Uint32 | import("../enum").Type.Uint64>>; } /** @ignore */ export declare class CountByResult<T extends DataType = any, TCount extends Int = Int> extends Table<{ values: T; counts: TCount; }> { constructor(values: Vector<T>, counts: V<TCount>); toJSON(): Object; } /** @ignore */ export declare class FilteredDataFrame<T extends { [key: string]: DataType; } = any> extends DataFrame<T> { private _predicate; constructor(batches: RecordBatch<T>[], predicate: Predicate); scan(next: NextFunc, bind?: BindFunc): void; scanReverse(next: NextFunc, bind?: BindFunc): void; count(): number; [Symbol.iterator](): IterableIterator<Struct<T>['TValue']>; filter(predicate: Predicate): FilteredDataFrame<T>; countBy(name: Col | string): CountByResult<any, Int<import("../enum").Type.Int | import("../enum").Type.Int8 | import("../enum").Type.Int16 | import("../enum").Type.Int32 | import("../enum").Type.Int64 | import("../enum").Type.Uint8 | import("../enum").Type.Uint16 | import("../enum").Type.Uint32 | import("../enum").Type.Uint64>>; }