import pandas as pd
import numpy as np
# List of Tuples
employees = [('Shubham', 'Data Scientist', 'Mumbai', 5),
('Riti', 'Data Scientist', 'Delhi' , 7),
('Shanky', 'Program Manager', 'Delhi' , 2),
('Shreya', 'Graphic Designer', 'Mumbai' , 2),
('Aadi', 'Data Scientist', 'Mumbai', 11)]
# Create a DataFrame object from list of tuples
df = pd.DataFrame(employees,
columns=['Name', 'Designation', 'City', 'Experience'],
index=[0, 1, 2, 3, 4])
#grouped = df.groupby(['City' ,'Designation'])
grouped = df.groupby(['City'])
print(grouped.size())
print("list grouped groups keys",list(grouped.groups.keys())[1])
group_values = grouped.get_group(list(grouped.groups.keys())[1])
print("group values \n", group_values)
#print("group values \n", group_values.iloc[[0]])
"""
class GetGroup:
def __init__(self,grouped_df,Nth_group):
self.grouped_df = grouped_df
self.Nth_group = Nth_group
self.group_name = list(self.grouped_df.groups.keys())[self.Nth_group]
self.group_vlue = self.grouped_df.get_group(list(self.grouped_df.groups.keys())[self.Nth_group])
"""
""" def Return_group_name (self):#self = grouped_df,Nth_group
return list(self.grouped_df.groups.keys())[self.Nth_group]
def Return_group_values(self):
return self.grouped_df.get_group(list(self.grouped_df.groups.keys())[self.Nth_group])
"""
def Get_groupNthGroup(grouped_df,Nth_group):
Get_groupNthGroup.group_name = list(grouped_df.groups.keys())[Nth_group]
Get_groupNthGroup.group_vlue = grouped_df.get_group(list(grouped_df.groups.keys())[Nth_group])
Get_groupNthGroup.value=10
Get_groupNthGroup(grouped,0)
print(Get_groupNthGroup.group_name)