在 DataFrame 中,您可能会遇到包含需要分组在一起的连续值的列。例如,考虑以下列的值:
[1, 1, -1, 1, -1, -1]
要有效地将这些值分组到所需的组中,例如:
[1,1] [-1] [1] [-1, -1]
使用 Pandas 库执行以下步骤:
您可以利用自定义系列来实现此分组。方法如下:
import pandas as pd # Create sample DataFrame df = pd.DataFrame({'a': [1, 1, -1, 1, -1, -1]}) # Use ne() and cumsum() to create grouping indicator ind = df['a'].ne(df['a'].shift()).cumsum() # Group by this indicator for i, g in df.groupby(ind): # Print grouping key print(i) # Print rows in group print(g) # Convert values to list for display print(g.a.tolist())
此代码将输出所需的分组和值:
1 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]
以上是如何有效地将 Pandas DataFrame 列中的连续值分组?的详细内容。更多信息请关注PHP中文网其他相关文章!