Groupby.get_groups

Getting Started API Reference Contributing Guide Github
Search…
Groupby.get_groups
Obtain the data for each element of the groupby column
danfo.Groupby.get_groups(key)      [source]
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!
Groupby
Groupby.col
Last modified 1yr ago
Copy link