pandas在groupby之后选择行

import pandas as pd

# 创建DataFrame
data = {'Name': ['John', 'Mike', 'Sarah', 'Mike', 'John'],
        'Age': [28, 32, 25, 32, 28],
        'City': ['New York', 'Los Angeles', 'Chicago', 'Los Angeles', 'New York']}
df = pd.DataFrame(data)

# 按照Name列进行分组
grouped = df.groupby('Name')

# 获取Name为John的行数据
john_data = grouped.get_group('John')
print(john_data)

import pandas as pd

# 创建DataFrame
data = {'Name': ['John', 'Mike', 'Sarah', 'Mike', 'John'],
        'Age': [28, 32, 25, 32, 28],
        'City': ['New York', 'Los Angeles', 'Chicago', 'Los Angeles', 'New York']}
df = pd.DataFrame(data)

# 按照Name列进行分组
grouped = df.groupby('Name')

# 选择Name为John的行数据
john_data = grouped['Name'].apply(lambda x: x == 'John')
print(df[john_data])