pandas 實現某一列分組,其他列合并成list

更新時間：2021年03月26日 11:13:48 作者：lxp198837

這篇文章主要介紹了pandas 實現某一列分組,其他列合并成list的案例。具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

pandas列轉換為字典，但將相同第一列（鍵）的所有值合并為一個鍵

形式一：

import pandas as pd 
# data
data = pd.DataFrame({'column1':['key1','key1','key2','key2'],
    'column2':['value1','value2','value3','value3']})
print(data) 
# Grouped dict
data_dict = data.groupby('column1').column2.apply(list).to_dict() 
print(data_dict)

輸出結果：

 column1 column2
0  key1 value1
1  key1 value2
2  key2 value3
3  key2 value3 
{'key1': ['value1', 'value2'], 'key2': ['value3', 'value3']}

形式二：

import pandas as pd
# data
df = pd.DataFrame({'column1':['key1','key1','key2','key2'],
    'column2':['value1','value2','value1','value2'],
    'column3':['value11','value11','value22','value22'],
    'column4':['value44','value44','value55','value55']}) 
# Grouped dict
data_dict = df.groupby('column1').apply(lambda x: {col:x[col].tolist() for col in x.columns if col != 'column2'}).to_dict()
print(data_dict) 
data_dict2 = df.groupby('column1').apply(lambda x: {col:x[col].tolist()[0] if col != 'column2' else x[col].tolist() for col in x.columns}).to_dict()
print(data_dict2)

輸出結果：

#data_dict
{
  'key1': {
    'column1': ['key1', 'key1'], 
    'column3': ['value11', 'value11'], 
    'column4': ['value44', 'value44']
  }, 
  'key2': {
    'column1': ['key2', 'key2'], 
    'column3': ['value22', 'value22'], 
    'column4': ['value55', 'value55']
  }
}
#data_dict2
{
  'key1': {
    'column1': 'key1', 
    'column2': ['value1', 'value2'], 
    'column3': 'value11', 
    'column4': 'value44'
  }, 
  'key2': {
    'column1': 'key2', 
    'column2': ['value1', 'value2'], 
    'column3': 'value22', 
    'column4': 'value55'
  }
}

補充：pandas中，利用groupby分組后，對字符串字段進行合并拼接

在pandas里對于數值字段而言，groupby后可以用sum()、max()等方法進行簡單的處理，對于字符串字段，如果把它們的值拼接在一起，可以用使用 str.cat() 和 lamda 方法。

如，將下面表格中的內容，對skill字段按照id進行分組合并

實現代碼：

import pandas as pd
file_name='test.xlsx'
df=pd.read_excel(file_name)
data=df.groupby('id')['skill'].apply(lambda x:x.str.cat(sep=':')).reset_index()
print(data)

效果如下：

另，數據處理時，常常需要將某一列進行拆分，分列，替換等，相關的函數有str.split()、str.extract()、str.replace().

以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章:

Python搭建代理IP池實現接口設置與整體調度
這篇文章主要介紹了Python搭建代理IP池實現接口設置與整體調度，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-10-10
Python字符串逆序輸出的實例講解
今天小編就為大家分享一篇關于Python字符串逆序輸出的實例講解，小編覺得內容挺不錯的，現在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧
2019-02-02
Python實現RLE格式與PNG格式互轉
在機器視覺領域的深度學習中，很多數據集的標注文件使用RLE的格式。但是神經網絡的輸入一定是一張圖片，為此必須把RLE格式的文件轉變?yōu)閳D像格式。本文將利用Python實現RLE格式與PNG格式互轉，感興趣的可以了解一下
2022-08-08
使用 Visual Studio Code（VSCode）搭建簡單的Python+Djan
這篇文章主要介紹了使用 Visual Studio Code（VSCode）搭建簡單的Python+Django開發(fā)環(huán)境的方法步驟，小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-12-12
Python接口自動化淺析數據驅動原理
這篇文章主要介紹了Python接口自動化淺析數據驅動原理,文中會詳細描述怎樣使用openpyxl模塊操作excel及結合ddt來實現數據驅動，有需要的朋友可以參考下
2021-08-08
一文詳解pygame.sprite的精靈碰撞
精靈其實在一個游戲程序中,精靈本質指的是一張張小尺寸的圖片,比如游戲中的各種道具、人物、場景裝飾等,它們都可以看做成一張張小的“精靈”圖,下面這篇文章主要給大家介紹了關于pygame.sprite精靈碰撞的相關資料,需要的朋友可以參考下
2023-01-01
python開啟多個子進程并行運行的方法
這篇文章主要介紹了python開啟多個子進程并行運行的方法,涉及Python進程操作的相關技巧,具有一定參考借鑒價值,需要的朋友可以參考下
2015-04-04
vscode寫python時的代碼錯誤提醒和自動格式化的方法
這篇文章主要介紹了vscode寫python時的代碼錯誤提醒和自動格式化的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-05-05
Python面向對象程序設計之類和對象、實例變量、類變量用法分析
這篇文章主要介紹了Python面向對象程序設計之類和對象、實例變量、類變量用法,結合實例形式分析了Python面向對象程序設計中類和對象、實例變量、類變量具體功能、原理、使用方法與操作注意事項,需要的朋友可以參考下
2020-03-03
Pytorch?使用Google?Colab訓練神經網絡深度學習
本文以VOC數據集為例，因此在訓練的時候沒有修改classes_path等，如果是訓練自己的數據集，各位一定要注意修改classes_path等其它參數
2022-04-04