@jrc03c/js-math-tools/src/dataframe/df-drop.mjs

some math tools for JS

import { assert } from "../assert.mjs"
import { forEach } from "../for-each.mjs"
import { isArray } from "../is-array.mjs"
import { isNumber } from "../is-number.mjs"
import { isString } from "../is-string.mjs"
import { isUndefined } from "../is-undefined.mjs"
import { shape } from "../shape.mjs"

function dfDrop(DataFrame, Series, df, rows, cols) {
  if (isUndefined(rows)) rows = []
  if (isUndefined(cols)) cols = []
  if (isString(rows) || isNumber(rows)) rows = [rows]
  if (isString(cols) || isNumber(cols)) cols = [cols]

  assert(
    isArray(rows),
    "The `drop` method only works on 1-dimensional arrays of numerical indices and/or strings.",
  )

  assert(
    isArray(cols),
    "The `drop` method only works on 1-dimensional arrays of numerical indices and/or strings.",
  )

  assert(
    shape(rows).length === 1,
    "The `drop` method only works on 1-dimensional arrays of numerical indices and/or strings.",
  )

  assert(
    shape(cols).length === 1,
    "The `drop` method only works on 1-dimensional arrays of numerical indices and/or strings.",
  )

  let outIndex, outColumns

  forEach(df.index, (row, i) => {
    if (rows.indexOf(row) < 0 && rows.indexOf(i) < 0) {
      if (!outIndex) outIndex = []
      outIndex.push(row)
    }
  })

  forEach(df.columns, (col, i) => {
    if (cols.indexOf(col) < 0 && cols.indexOf(i) < 0) {
      if (!outColumns) outColumns = []
      outColumns.push(col)
    }
  })

  let out = df.get(outIndex, outColumns)

  if (out instanceof Series) {
    let temp = new DataFrame()
    temp = temp.assign(out)
    if (df.index.indexOf(out.name) > -1) temp = temp.transpose()
    out = temp
  }

  return out
}

export { dfDrop }