我有一个100196行x8列的数据帧.
我希望得到类似如下所示的输出
列2到列5具有相同的数据，因为ID相同.我只想有所有的设计代码在一个单元格作为新行.我知道我们可以使用GROUPBY和AGG函数，但我只能添加一列.
有谁能建议一下，如何解决这个问题？
我有一个100196行x8列的数据帧.
我希望得到类似如下所示的输出
列2到列5具有相同的数据，因为ID相同.我只想有所有的设计代码在一个单元格作为新行.我知道我们可以使用GROUPBY和AGG函数，但我只能添加一列.
有谁能建议一下，如何解决这个问题？
我已经根据您的示例创建了一个模拟数据集:
import pandas as pd
import numpy as np
df = pd.DataFrame({
"ID": ["1", "1", "2", "3", "3", "3"],
"Data 1": ["data11", "data11", "data12", "data13", "data13", "data13"],
"Data 2": ["data21", "data21", "data22", "data23", "data23", "data23"],
"Design Code": ["a", "b", "c", "d", "e", "f"]
})
您可以对GROUPBY中的不同列应用不同的聚合，以获得预期的结果，例如:
df \
.groupby("ID", as_index=False) \
.agg({
"Data 1": min,
"Data 2": min,
"Design Code": lambda x: "\n".join(np.unique(x)),
})
输出:
ID Data 1 Data 2 Design Code
0 1 data11 data21 a\nb
1 2 data12 data22 c
2 3 data13 data23 d\ne\nf
至于Design Code带换行符的显示，如果你使用的是Jupyter笔记本，你可以参考https://stackoverflow.com/a/46326725/22052558.