Pandas DataFrame：如何组合连续值

Question

Pandas DataFrame：如何组合连续值我在DataFrame中有一个包含值的列：[1, 1, -1, 1, -1, -1]我该如何将它们分组呢？[1,1] [-1] [1] [-1, -1]

Answer 1

您可以groupby按自定义使用Series：

df = pd.DataFrame({'a': [1, 1, -1, 1, -1, -1]})

print (df)

   a

0  1

1  1

2 -1

3  1

4 -1

5 -1

print ((df.a != df.a.shift()).cumsum())

0    1

1    1

2    2

3    3

4    4

5    4

Name: a, dtype: int32

for i, g in df.groupby([(df.a != df.a.shift()).cumsum()]):

    print (i)

    print (g)

    print (g.a.tolist())

   a

0  1

1  1

[1, 1]

2

   a

2 -1

[-1]

3

   a

3  1

[1]

4

   a

4 -1

5 -1

[-1, -1]

Answer 2

使用groupby从itertools来自杰斯数据

from itertools import groupby[ list(group) for key, group in groupby(df.a.values.tolist())]Out[361]: [[1, 1], [-1], [1], [-1, -1]]