Class DataFrame

src/danfojs-base/core/frame.ts:48–3425

 * @param options.config General configuration object for extending or setting NDframe behavior.      
 */
export default class DataFrame extends NDframe implements DataFrameInterface {
    [key: string]: any
    constructor(data: any, options: BaseDataOptionType = {}) {
        const { index, columns, dtypes, config } = options;
        super({ data, index, columns, dtypes, config, isSeries: false });
        this.$setInternalColumnDataProperty();
    }

    /**
     * Maps all column names to their corresponding data, and return them as Series objects.
     * This makes column subsetting works. E.g this can work ==> `df["col1"]`
     * @param column Optional, a single column name to map
     */
    private $setInternalColumnDataProperty(column?: string) {
        const self = this;
        if (column && typeof column === "string") {
            Object.defineProperty(self, column, {
                get() {
                    return self.$getColumnData(column)
                },
                set(arr: ArrayType1D | Series) {
                    self.$setColumnData(column, arr);
                }
            })
        } else {
            const columns = this.columns;
            for (let i = 0; i < columns.length; i++) {
                const column = columns[i];
                Object.defineProperty(this, column, {
                    get() {
                        return self.$getColumnData(column)
                    },
                    set(arr: ArrayType1D | Series) {
                        self.$setColumnData(column, arr);
                    }
                })
            }
        }

    }

    /**
     * Returns the column data from the DataFrame by column name. 
     * @param column column name to get the column data
     * @param returnSeries Whether to return the data in series format or not. Defaults to true
     */
    private $getColumnData(column: string, returnSeries: boolean = true) {
        const columnIndex = this.columns.indexOf(column)

        if (columnIndex == -1) {
            ErrorThrower.throwColumnNotFoundError(this)
        }

        const dtypes = [this.$dtypes[columnIndex]]
        const index = [...this.$index]
        const columns = [column]
        const config = { ...this.$config }