Pandas分组生成唯一ID方法

小伙伴们有没有觉得学习文章很有意思？有意思就对了！今天就给大家带来《Pandas分组生成唯一ID技巧》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

一、背景与挑战

在数据处理和分析中，我们经常需要为数据集中的记录生成唯一的标识符。当这些标识符需要基于多列信息，并且要求在某一列（如ID）的分组内，根据另一列（如Name）的出现顺序或唯一性来生成时，问题会变得复杂。例如，给定一个DataFrame，其中包含Name和ID两列：

我们的目标是创建一个新的ID_new列，其格式为原始ID_后缀，其中后缀表示该Name在对应ID组内的唯一实例序号。例如，ID=1的第一个A和第二个A都应对应1_1，而ID=3的C是第一个Name，D是第二个Name，则分别对应3_1和3_2。

传统的Pandas方法如groupby().ngroup()虽然可以生成组内序号，但它为每个组生成一个唯一的整数，不适用于我们这种需要保留原始ID并添加后缀的场景。而groupby().cumcount()则会为每个组内的行生成递增序号，导致对于相同的Name在同一ID组内产生不同的后缀（例如，ID=1的两个A会分别得到1_1和1_2），这不符合我们的“唯一实例”要求。此外，对于拥有数万甚至数十万个ID的大规模数据集，某些方法可能会面临性能瓶颈。

二、解决方案：结合 factorize 与 groupby().transform()

解决此类问题的有效方法是利用pandas.factorize函数在groupby().transform()的上下文中使用。factorize函数能够将数组中的唯一值映射为整数编码，这正是我们所需的“唯一实例序号”。transform()则确保了操作结果能够正确地对齐回原始DataFrame的索引。

2.1 核心原理

1. 分组 (groupby('ID')): 首先，根据ID列对DataFrame进行分组。
2. 组内应用函数 (transform(f)): 对于每个ID组，我们只关注Name列。transform()函数会将一个函数应用于每个组的Name列，并返回一个与原始DataFrame具有相同索引的Series。
3. factorize 生成后缀: 在transform中应用的函数f会调用pd.factorize(x)[0] + 1。
   • pd.factorize(x)会返回一个元组，其中第一个元素是整数数组，代表x中唯一值的编码。例如，['A', 'B', 'A', 'C']经过factorize可能得到[0, 1, 0, 2]。
   • [0]用于提取这个整数数组。
   • + 1是为了让序号从1开始，而非0。
   • 这样，对于每个ID组，factorize会为该组内出现的每个唯一Name分配一个唯一的、从1开始的整数后缀。
4. 拼接 (str.cat): 最后，将原始的ID列（转换为字符串）与生成的后缀（也转换为字符串）通过分隔符_进行拼接，形成新的ID_new列。

2.2 示例代码

让我们通过一个具体的例子来演示这个过程。

import pandas as pd

# 准备示例数据
data = {
    'Name': ['A', 'B', 'A', 'C', 'B', 'D', 'E', 'F'],
    'ID': [1, 2, 1, 3, 2, 3, 1, 2]
}
df = pd.DataFrame(data)

print("原始 DataFrame:")
print(df)

# 定义一个lambda函数，用于在每个组内对Name列进行factorize编码
# factorize返回的第一个元素是编码数组，我们将其加1使其从1开始计数
f = lambda x: pd.factorize(x)[0] + 1

# 使用groupby().transform()在每个ID组内生成Name的唯一序号
# transform确保了结果Series的索引与原始DataFrame对齐
s = df.groupby('ID')['Name'].transform(f).astype(str)

# 将原始ID列转换为字符串，并与生成的后缀进行拼接
df['ID_new'] = df['ID'].astype(str).str.cat(s, sep='_')

print("\n生成 'ID_new' 列后的 DataFrame:")
print(df)

输出结果:

原始 DataFrame:
  Name  ID
0    A   1
1    B   2
2    A   1
3    C   3
4    B   2
5    D   3
6    E   1
7    F   2

生成 'ID_new' 列后的 DataFrame:
  Name  ID ID_new
0    A   1    1_1
1    B   2    2_1
2    A   1    1_1
3    C   3    3_1
4    B   2    2_1
5    D   3    3_2
6    E   1    1_2
7    F   2    2_2

从结果可以看出，ID=1的两个A都得到了1_1，而ID=3的C得到3_1，D得到3_2，这完全符合我们的预期。ID=1的E由于是ID=1组内第一次出现E，因此得到1_2（因为A已经占用了1）。

三、注意事项与性能考量

1. 数据类型转换: 在拼接字符串之前，务必将ID列和生成的后缀Series都转换为字符串类型（astype(str)），否则str.cat方法可能无法正常工作。
2. 性能优势: pd.factorize在底层使用优化的C语言实现，效率非常高。结合groupby().transform()，这种方法能够高效地处理大规模数据集，避免了Python循环的开销，尤其适用于拥有大量分组和记录的场景。相比于ngroup()或cumcount()在特定逻辑下的局限性，此方法提供了更精确和高效的解决方案。
3. 灵活性: 分隔符sep='_'可以根据需求进行调整。如果需要更复杂的后缀生成逻辑，也可以修改lambda函数中的factorize部分，或者替换为其他自定义函数。
4. factorize的顺序: factorize的编码顺序取决于元素在输入Series中第一次出现的顺序。这意味着如果ID=1组中先出现A，再出现E，那么A会得到1，E会得到2。如果数据顺序发生变化，编码结果可能也会变化，但对于“唯一实例”的语义是保持一致的。

四、总结

本文介绍了一种在Pandas DataFrame中高效生成基于分组的唯一组合ID的专业方法。通过巧妙地结合pandas.factorize函数与groupby().transform()操作，我们能够克服传统方法的局限性，为每个分组内的独特实例生成正确的递增后缀，并将其与原始ID进行拼接。这种方法不仅逻辑清晰、易于理解，而且在处理大规模数据集时表现出卓越的性能，是数据科学家和工程师在处理类似需求时的理想选择。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Pandas分组生成唯一ID方法》文章吧，也可关注golang学习网公众号了解相关技术文章。