Skip to content

Commit 9d8c6fb

Browse files
committed
update groupby concatGroups to make use of atIndex in addColumn
1 parent c07f100 commit 9d8c6fb

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

src/danfojs-base/aggregators/groupby.ts

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -376,15 +376,17 @@ export default class Groupby {
376376
copyDf = new DataFrame([values.values], {columns: columns })
377377
}
378378
let len = copyDf.shape[0]
379-
for(let key1 in this.keyCol){
379+
let key1: any;
380+
for(key1 in this.keyCol){
381+
380382
let keyName = this.keyCol[key1] as string
381383
let keyValue = this.keyToValue[key][key1]
382384
let dfValue = Array(len).fill(keyValue)
383385
if (this.groupColNames) {
384-
copyDf.addColumn(keyName, dfValue, {inplace: true})
386+
copyDf.addColumn(keyName, dfValue, {inplace: true, atIndex: key1 as number })
385387
}
386388
else {
387-
copyDf.addColumn(`${keyName}_Group`, dfValue, {inplace: true})
389+
copyDf.addColumn(`${keyName}_Group`, dfValue, {inplace: true, atIndex: key1 as number })
388390
}
389391

390392
}

0 commit comments

Comments
 (0)