基于两个唯一列值pandas创建集群的函数

import pandas as pd

def create_cluster(df, column1, column2):
    # 使用groupby函数根据两个列值进行分组
    grouped = df.groupby([column1, column2])
    
    # 遍历每个分组并输出结果
    for name, group in grouped:
        print("Cluster:", name)
        print(group)
        print("--------------------")

# 示例用法
# 创建一个包含两个唯一列值的DataFrame
data = {'Column1': ['A', 'A', 'B', 'B', 'C'],
        'Column2': [1, 2, 1, 2, 1],
        'Value': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 调用函数创建集群
create_cluster(df, 'Column1', 'Column2')