在Dataframe列中保留重复项的第一个和最后一个条目

import pandas as pd

# 读取数据并创建Dataframe对象
data = {'col1': [1, 2, 2, 3, 4, 4, 5],
        'col2': ['a', 'b', 'b', 'c', 'd', 'd', 'e']}
df = pd.DataFrame(data)

# 标记重复项
df['is_duplicate'] = df.duplicated(subset='col1', keep=False)

# 保留第一个和最后一个重复项
df_first_last = df.drop_duplicates(subset='col1', keep='first')
df_first_last = df_first_last.append(df.drop_duplicates(subset='col1', keep='last'))

# 打印结果
print(df_first_last)