如何迭代组

Question

我正在尝试迭代组（由 Pandas 中的 group.by 生成）以应用函数（在 MatPlotLib 中创建图表）并为 DataFrame 中的每个组获取结果。我想做这样的事情，但我知道有更好/更实用的方法：

import pandas as pd
DF = pd.DataFrame({'COL1': ['A', 'A','B','B'], 
                   'COL2' : [1,1,2,2],
                   'COL3' : [2,3,4,5]})


unique_list = set(df['COL1'])
for group in unique_list:
    if group == df['COL1']:
       <apply function to df['COL3'] 
       one group at a time, 
       starting with df['COL1'] == 'A'> 

Answer 1

groupby 函数的结果是一个 pandas 数据框或系列。您可以使用应用功能。见下例：

groupedDF = DF.groupby(['COL1']).sum()

#Let's assume we want apply below function
def square(x):
    return x['COL2']*x['COL2']

#Below line will return the results with applied function
groupedDF.apply(square,axis=1)

#Let's assume Returned value is series
groupedSeries = DF.groupby(['COL1'])['COL2'].sum()

#Let's assume we want apply below function
def square(x):
    return x*x

#Below line will return the result
groupedSeries.apply(square)