如何在Pandas中找到重复实例的第一个实例并将其标记在不同的列上？

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 找到重复实例
duplicates = df.duplicated()

# 将重复实例标记在不同的列上
df['IsDuplicate'] = duplicates.where(duplicates, False)
df['FirstInstance'] = df['A'].where(duplicates, pd.NA)

print(df)

   A  B  IsDuplicate FirstInstance
0  1  1        False          <NA>
1  2  2        False          <NA>
2  3  3        False          <NA>
3  4  4        False          <NA>
4  5  5        False          <NA>