Блог

Я работаю над обширной обработкой данных в nodejs. Я нашел danfojs как хорошую альтернативу python Pandas. Но сталкивается с некоторыми недостатками функциональности по сравнению с Pandas.

Как разработать функциональность pandas isin() в Danfojs?

Пример:

У меня есть приведенный ниже фрейм данных:

И приведенная ниже серия:

Я хочу получить те строки фрейма данных, идентификатор которых существует в серии

Я не думаю, что существует эквивалент pandas ‘isin’, но я нашел обходной путь для решения этой проблемы

 const dfd = require("danfojs-node")
const data = [
    ['asefwc', 'Abdullah', 'Cumilla'],
    ['wefcss', 'Khairul', 'Jashore'],
    ['erfegf',  'Jaman', 'Magura']
]

const columns = ['id', 'name', 'address']

let df = new dfd.DataFrame(data, { columns })

let ids = df['id'].values // get all the column values

let allRowsToInclude = []

// idsToInclude is your Series
idsToInclude.forEach((idToInclude) => {
  const rowsToInclude = ids.flatMap((id, idx) => id === idToInclude ? idx : [])
  allRowsToInclude = allRowsToInclude.push(... rowsToInclude)
}

// creates a new df with filtered rows
let dfFinal = df.loc({rows: allRowsToInclude}) 
 

Я думаю, что стоит изучить dfd.DataFrame.merge, dfd.DataFrame.merge, по сути, выбирая конкретный вариант соединения, вы фильтруете данные определенным образом.