给时间序列中连续唯一值的序列分配分组ID

以下是一个使用Python解决给时间序列中连续唯一值的序列分配分组ID的示例代码：

import pandas as pd

# 创建一个示例时间序列
data = pd.Series(['A', 'A', 'B', 'C', 'C', 'C', 'D', 'D', 'D', 'D'])
print("原始序列:")
print(data)

# 使用shift函数和cumsum函数创建一个布尔序列，表示每个唯一值的起始位置
start_of_group = data.ne(data.shift())

# 使用cumsum函数为每个分组分配一个唯一的ID
group_id = start_of_group.cumsum()

# 将分组ID添加为新的一列
data_with_group_id = pd.DataFrame({'value': data, 'group_id': group_id})
print("\n分配分组ID后的序列:")
print(data_with_group_id)

输出结果为：

原始序列:
0    A
1    A
2    B
3    C
4    C
5    C
6    D
7    D
8    D
9    D
dtype: object

分配分组ID后的序列:
  value  group_id
0     A         1
1     A         1
2     B         2
3     C         3
4     C         3
5     C         3
6     D         4
7     D         4
8     D         4
9     D         4

在上面的示例代码中，我们首先创建了一个示例时间序列。然后使用shift函数和cumsum函数来创建一个布尔序列，表示每个唯一值的起始位置。最后，使用cumsum函数为每个分组分配一个唯一的ID，并将分组ID添加为新的一列。