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)   a0  11  12 -13  14 -15 -1print ((df.a != df.a.shift()).cumsum())0    11    12    23    34    45    4Name: a, dtype: int32for i, g in df.groupby([(df.a != df.a.shift()).cumsum()]):    print (i)    print (g)    print (g.a.tolist())   a0  11  1[1, 1]2   a2 -1[-1]3   a3  1[1]4   a4 -15 -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]]