您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:pandas 实现某一列分组,其他列合并成list

51自学网 2021-10-30 22:47:15
  python
这篇教程pandas 实现某一列分组,其他列合并成list写得很实用,希望能帮到您。

pandas列转换为字典,但将相同第一列(键)的所有值合并为一个键

形式一:

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

输出结果:

 column1 column20  key1 value11  key1 value22  key2 value33  key2 value3 {'key1': ['value1', 'value2'], 'key2': ['value3', 'value3']}

形式二:

import pandas as pd# datadf = pd.DataFrame({'column1':['key1','key1','key2','key2'],    'column2':['value1','value2','value1','value2'],    'column3':['value11','value11','value22','value22'],    'column4':['value44','value44','value55','value55']}) # Grouped dictdata_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 pdfile_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().

以上为个人经验,希望能给大家一个参考,也希望大家多多支持51zixue.net。如有错误或未考虑完全的地方,望不吝赐教。


pandas 实现分组后取第N行
利用pandas按日期做分组运算的操作
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。