在单独的数据框中收集 ID

Question

通过 id 对我的数据进行分组后，我陷入了困境。 现在我想用相同的 id 收集它们的完整信息。

当前 => 在使用 group by id 后我得到了这个结果。 预期 => 具有所有信息的每个相同 id，如图所示

dfs=pd.read_excel('tns1.xlsx')
grp = dfs.groupby('entity_id')
da  = grp.groups
for entity_id,show in grp:
  print(show)
  print(da)

现在我也有预期的输出，但问题是我正在正确地提取它们然后使用下面的循环在 excel 中写入我正在尝试提取但我不认为它除了运行数千次之外还可以正确提取因为有很多行

Answer 1

因此，您可以收集要列出的个人数据，然后将其写入 csv。

用于说明的数据设置：

import pandas as pd
from io import StringIO
raw_data="""
entity_id,Name,Status,Date
2244,Abhi,Active,10-06-2021
2244,Abhi2,Blocked,10-06-2021
6666,other1,Blocked,10-06-2021
6666,other,Active,10-06-2021
"""
dfs=pd.read_csv(StringIO(raw_data))

将所有 dfs 附加到列表中：

grp = dfs.groupby('entity_id')
da  = grp.groups
groups_as_list=[]

for entity_id,show in grp:
    groups_as_list.append(show)# append all dfs in a list

然后将数据帧列表写入 csv。

with open("my_results.csv","w") as f:# open file for write
    for each_df in groups_as_list:
            each_df.to_csv(f,index=False)
            f.write("\n") # write empty line to csv

所以结果将如下所示：（您可以根据需要进一步调整）