Pandas是一種開源Python庫，用于數(shù)據(jù)操作和數(shù)據(jù)分析。其中的groupby函數(shù)可以將數(shù)據(jù)按指定的列或條件進(jìn)行分組，這是數(shù)據(jù)分析中非常常用的功能之一。在pandas分組后，我們可能需要對每個分組進(jìn)行遍歷處理，例如進(jìn)行統(tǒng)計(jì)、計(jì)算、篩選等操作。本文將介紹如何在pandas分組后對數(shù)據(jù)進(jìn)行遍歷處理。

1. 分組方法

在pandas中，可以使用groupby方法對數(shù)據(jù)進(jìn)行分組，該函數(shù)返回一個GroupBy對象。GroupBy對象是一個非常強(qiáng)大的對象，它包含了很多有用的方法，可以用來對數(shù)據(jù)進(jìn)行聚合、轉(zhuǎn)換、過濾等操作。下面是一個示例，展示如何通過groupby方法分組數(shù)據(jù)：

import pandas as pd # 創(chuàng)建一個DataFrame data = {'name': ['小明', '小紅', '小明', '小紅'], 'age': [18, 20, 19, 21], 'city': ['北京', '上海', '廣州', '深圳']}

df = pd.DataFrame(data) # 按照name列進(jìn)行分組 grouped = df.groupby('name')

2. 遍歷分組

對于groupby方法分組后的數(shù)據(jù)，我們可以使用for循環(huán)來遍歷每個分組。在每次循環(huán)中，我們將得到一個元組，其中第一個元素是分組的名稱（也就是按照哪個列進(jìn)行分組），第二個元素是一個DataFrame對象，包含了該分組的所有數(shù)據(jù)。下面是一個示例：

# 遍歷每個分組 for name, group in grouped: print(name) print(group)

輸出結(jié)果如下：

小明
   name  age city
0   小明   18  北京
2   小明   19  廣州

小紅
   name  age city
1   小紅   20  上海
3   小紅   21  深圳

3. 分組統(tǒng)計(jì)

在遍歷分組后，我們可以對每個分組進(jìn)行統(tǒng)計(jì)計(jì)算。例如，我們可以計(jì)算每個分組的平均值、最大值、最小值等。在pandas中，我們可以使用agg函數(shù)來實(shí)現(xiàn)這些操作。agg函數(shù)接受一個字典參數(shù)，其中鍵表示要計(jì)算的列名稱，值表示要進(jìn)行的計(jì)算操作。下面是一個示例：

# 計(jì)算每個分組的平均年齡和最大年齡 result = grouped.agg({'age': ['mean', 'max']}) print(result)

輸出結(jié)果如下：

age    
         mean max
name            
小明    18.500  19
小紅    20.500  21

4. 分組篩選

在遍歷分組后，我們還可以根據(jù)條件篩選分組。例如，我們可以只選擇年齡大于等于20歲的分組。在pandas中，我們可以使用filter函數(shù)來實(shí)現(xiàn)這個操作。filter函數(shù)接受一個函數(shù)參數(shù)，該函數(shù)應(yīng)該返回一個布爾值，表示是否選擇該分組。下面是一個示例：

# 篩選年齡大于等于20歲的分組 def filter_func(x): return x['age'].mean() >= 20 result = grouped.filter(filter_func) print(result)

輸出結(jié)果如下：

name  age city
1   小紅   20  上海
3   小紅   21  深圳

5. 分組轉(zhuǎn)換

在遍歷分組后，我們還可以對每個分組進(jìn)行轉(zhuǎn)換。例如，我們可以將每個分組的年齡減去該分組的平均年齡。在pandas中，我們可以使用transform函數(shù)來實(shí)現(xiàn)這個操作。transform函數(shù)接受一個函數(shù)參數(shù)，該函數(shù)

應(yīng)該返回一個與分組大小相同的Series或DataFrame對象。下面是一個示例：

# 將每個分組的年齡減去平均年齡 def transform_func(x):
    x['age'] = x['age'] - x['age'].mean() return x

result = grouped.apply(transform_func) print(result)

輸出結(jié)果如下：

name   age city
0   小明 -0.500  北京
1   小紅  0.500  上海
2   小明  0.500  廣州
3   小紅  0.500  深圳

6. 總結(jié)

在本文中，我們介紹了如何在pandas分組后對數(shù)據(jù)進(jìn)行遍歷處理。首先，我們使用groupby方法對數(shù)據(jù)進(jìn)行分組。然后，我們可以使用for循環(huán)遍歷分組，并對每個分組進(jìn)行統(tǒng)計(jì)、篩選、轉(zhuǎn)換等操作。例如，我們可以使用agg函數(shù)計(jì)算每個分組的平均值、最大值等；使用filter函數(shù)根據(jù)條件選擇分組；使用transform函數(shù)對每個分組進(jìn)行轉(zhuǎn)換。這些操作非常有用，在實(shí)際的數(shù)據(jù)分析和處理中經(jīng)常會用到。

想快速入門Python數(shù)據(jù)分析？這門課程適合你！

如果你對Python數(shù)據(jù)分析感興趣，但不知從何入手，推薦你學(xué)習(xí)《山有木兮：Python數(shù)據(jù)分析極簡入門》。這門課程專為初學(xué)者設(shè)計(jì)，內(nèi)容簡潔易懂，手把手教你掌握Python數(shù)據(jù)分析的核心技能，助你輕松邁出數(shù)據(jù)分析的第一步。

學(xué)習(xí)入口：https://edu.cda.cn/goods/show/3429?targetId=5724&preview=0
開啟你的Python數(shù)據(jù)分析之旅，從入門到精通，只需一步！