Python 在pandas中列出对相同行的理解

我想知道是否有一种简洁的方法可以在存在相同行的情况下“折叠”熊猫数据帧。例如：

df=

我需要的是：

df_new = 

col_a     col_b
    a         1
    b    [2, 3]
    c         4
    d [5, 6, 7]

它肯定应该包括groupby

但如何有效地实现括号中的位，我感到困惑。

您可以将列表应用于列b：

---

groupby'col_a'。col_b.applylist应该可以工作。我相信这个问题的确是个愚蠢的想法。谢谢我应该删除这个问题吗？天哪，我从来没有觉得这么傻：

df_new = 

col_a     col_b
    a         1
    b    [2, 3]
    c         4
    d [5, 6, 7]

df_new = df.groupby('col_a').apply(....)

df.groupby('col_a')['col_b'].apply(list)

col_a
a          [1]
b       [2, 3]
c          [4]
d    [5, 6, 7]
Name: col_b, dtype: object

s = df.groupby('col_a')['col_b'].apply(list)
df['col_c'] = df['col_a'].map(s)

print(df)

col_a   col_b   col_c
0   a   1   [1]
1   b   2   [2, 3]
2   b   3   [2, 3]
3   c   4   [4]
4   d   5   [5, 6, 7]
5   d   6   [5, 6, 7]
6   d   7   [5, 6, 7]