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

1. 分組方法

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

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

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

2. 遍歷分組

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

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

輸出結果如下：

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

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

3. 分組統計

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

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

輸出結果如下：

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

4. 分組篩選

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

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

輸出結果如下：

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

5. 分組轉換

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

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

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

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

輸出結果如下：

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

6. 總結

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

想快速入門Python數據分析？這門課程適合你！

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

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