Groupby.get_groups
Obtain the data for each element of the groupby column
Parameters
Type
Description
default
key
Array
element of the groupby column
Returns: DataFrame
Example
Group the dataframe by column A and obtain the group belonging to the values in column A
Node
1
const dfd = require("danfojs-node")
2
3
let data ={'A': ['foo', 'bar', 'foo', 'bar',
4
'foo', 'bar', 'foo', 'foo'],
5
'B': ['one', 'one', 'two', 'three',
6
'two', 'two', 'one', 'three'],
7
'C': [1,3,2,4,5,2,6,7],
8
'D': [3,2,4,1,5,6,7,8]
9
}
10
11
let df = new dfd.DataFrame(data)
12
13
14
let grp = df.groupby(["A"])
15
16
grp.get_groups(["foo"]).print()
17
18
grp.get_groups(["bar"]).print()
Copied!
1
//get groups for key "foo"
2
Shape: (5,4)
3
4
╔═══╤═══════════════════╤═══════════════════╤═══════════════════╤═══════════════════╗
5
║ │ A │ B │ C │ D ║
6
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
7
║ 0 │ foo │ one │ 1 │ 3 ║
8
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
9
║ 1 │ foo │ two │ 2 │ 4 ║
10
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
11
║ 2 │ foo │ two │ 5 │ 5 ║
12
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
13
║ 3 │ foo │ one │ 6 │ 7 ║
14
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
15
║ 4 │ foo │ three │ 7 │ 8 ║
16
╚═══╧═══════════════════╧═══════════════════╧═══════════════════╧═══════════════════╝
17
18
//get groups for key "bar"
19
20
Shape: (3,4)
21
22
╔═══╤═══════════════════╤═══════════════════╤═══════════════════╤═══════════════════╗
23
║ │ A │ B │ C │ D ║
24
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
25
║ 0 │ bar │ one │ 3 │ 2 ║
26
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
27
║ 1 │ bar │ three │ 4 │ 1 ║
28
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
29
║ 2 │ bar │ two │ 2 │ 6 ║
30
╚═══╧═══════════════════╧═══════════════════╧═══════════════════╧═══════════════════╝
Copied!
Group dataframe by two columns and obtain their groups. Since the dataframe is grouped by two columns we most specify two keys in the get_groups belonging to these two columns
Node
1
const dfd = require("danfojs-node")
2
3
let data ={'A': ['foo', 'bar', 'foo', 'bar',
4
'foo', 'bar', 'foo', 'foo'],
5
'B': ['one', 'one', 'two', 'three',
6
'two', 'two', 'one', 'three'],
7
'C': [1,3,2,4,5,2,6,7],
8
'D': [3,2,4,1,5,6,7,8]
9
}
10
11
let df = new dfd.DataFrame(data)
12
13
14
let grp = df.groupby(["A","B"])
15
16
grp.get_groups(["foo","one"]).print()
17
18
grp.get_groups(["bar","one"]).print()
Copied!
1
//get_groups(["foo","one"]
2
3
4
Shape: (2,4)
5
6
╔═══╤═══════════════════╤═══════════════════╤═══════════════════╤═══════════════════╗
7
║ │ A │ B │ C │ D ║
8
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
9
║ 0 │ foo │ one │ 1 │ 3 ║
10
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
11
║ 1 │ foo │ one │ 6 │ 7 ║
12
╚═══╧═══════════════════╧═══════════════════╧═══════════════════╧═══════════════════╝
13
14
//get_groups(["bar","one"])
15
16
17
Shape: (1,4)
18
19
╔═══╤═══════════════════╤═══════════════════╤═══════════════════╤═══════════════════╗
20
║ │ A │ B │ C │ D ║
21
╟───┼───────────────────┼───────────────────┼───────────────────┼───────────────────╢
22
║ 0 │ bar │ one │ 3 │ 2 ║
23
╚═══╧═══════════════════╧═══════════════════╧═══════════════════╧═══════════════════╝
Copied!
Last modified 3mo ago
Copy link