python - 将几行与 Pandas 的定界符连接成一行

Question

我希望能够根据 ID 将多行字符串连接成一行。我使用库 pandas (python 3)。

val   id
Cat   1
Tiger 2
Ball  3
Bat   1
bill  2
dog   1

l = []
a = 0
while a < lendata:
    if df["id"][a] == 1:
        if a != 0:
            df["val"][tmp] = ' '.join(l)
            l = []
        tmp = a
        l.append(df["val"][a])
    else:
        l.append(df["val"][a])
    a += 1

它适用于循环。我需要这个结果，

val
Cat Tiger Ball
Bat bill
dog

不是分组依据

问题:您知道如何使用 pandas 函数来实现吗？谢谢。

Answer 1

留在 Pandas 中:

df['group'] = (df['id'] == 1).cumsum()
df.groupby('group')['val'].apply(' '.join).reset_index()

   id             val
0   1  Cat Tiger Ball
1   2        Bat bill
2   3             dog

第一行根据您的定义定义组。第二行是标准的groupby操作。