代码之家 › 专栏 › 技术社区 › The Great

拆分并存储数据帧,但名称基于特定列中的唯一值

group-by dataframe list pandas python

The Great · 技术社区 · 2 年前

我有一个如下的数据帧

data = pd.DataFrame({'email_id': ['abc@gmail.com;test1@gmail.com','abc@gmail.com;def@yahoo.com','abdc@gmail.com','ache@gmail.com','aqce@gmail.com','pqr@gmail.com','pqr@gmail.com'],
             'Dept_id': [21,23,25,26,28,29,31],
             'dept_name':['Science','Chemistry','Maths','Social','Physics','Botany','Zoology'],
             'location':['KOR','ASN','ANZ','IND','AUS','NZ','NZ']})

我想做以下事情

a) 基于唯一性拆分数据帧 email_id 电子邮件id 柱

b) 将分割数据存储在excel文件中。每个email\u id行将有一个文件。

c) 根据中存在的相应唯一值命名excel文件 dept_id dept_name 和 location

for k, v in data.groupby(['email_id']):
    dept_unique_ids = v['Dept_id'].unique()
    dept_unique_names = v['dept_name'].unique()
    location_unique = v['location'].unique()
    writer = pd.ExcelWriter(f"{k}.xlsx", engine='xlsxwriter')
    v.to_excel(writer,columns=col_list,sheet_name=f'{k}',index=False, startrow = 1)

电子邮件id 它被用作键。相反,我想根据 , 地方

email_id = pqr@gmail.com ,它们有两个唯一的dept\u ID,分别为29和31,其唯一的Depth\u名称为 Botany and Zoology 独特的位置是 NZ .

29_31_Botany_Zoology_NZ.xlsx

因此,我希望我的输出文件(每个唯一的电子邮件id行都有如下文件名)

更新-错误消息

1 回复 | 直到 2 年前

Corralien 2 年前

IIUC,您可以使用:

for k, v in data.groupby(['email_id']):
    dept_unique_ids = '_'.join(v['Dept_id'].astype(str).unique())
    dept_unique_names = '_'.join(v['dept_name'].unique())
    location_unique = '_'.join(v['location'].unique())
    filename = '_'.join([dept_unique_ids, dept_unique_names, location_unique])
    print(filename)

    # Use a context manager
    with pd.ExcelWriter(f"{filename}.xlsx", engine='xlsxwriter') as writer:
        v.to_excel(writer,columns=col_list,sheet_name=f'{k}',index=False, startrow=1)

输出:

23_Chemistry_ASN
21_Science_KOR
25_Maths_ANZ
26_Social_IND
28_Physics_AUS
29_31_Botany_Zoology_NZ

推荐文章

danial · 如何在多个字符串的每个位置找到最频繁的字符

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

shekharsabale · 从列表元素捕获子字符串

2 年前

The Great · 拆分并存储数据帧,但名称基于特定列中的唯一值

2 年前

Lakshya Kashyap · 如何先按数字降序排序,然后按字母升序排序

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

Klimt865 · Python中的列表列表

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

dekt · 获取数字列表Top N、next Top N和last Top N的索引

2 年前

FÄÅ ÛÅ · 字典、列表和字符串

2 年前