从dataframe列创建集合和计数字典

import pandas as pd

# 创建一个示例dataframe
data = {'column1': [1, 2, 3, 4, 5],
        'column2': ['a', 'b', 'c', 'd', 'e']}
df = pd.DataFrame(data)

# 选择需要的列
selected_columns = df[['column1', 'column2']]

# 创建集合
column1_set = set(selected_columns['column1'])
column2_set = set(selected_columns['column2'])

# 创建计数字典
column1_dict = selected_columns['column1'].value_counts().to_dict()
column2_dict = selected_columns['column2'].value_counts().to_dict()

print("column1集合:", column1_set)
print("column2集合:", column2_set)
print("column1计数字典:", column1_dict)
print("column2计数字典:", column2_dict)

column1集合: {1, 2, 3, 4, 5}
column2集合: {'c', 'd', 'e', 'b', 'a'}
column1计数字典: {1: 1, 2: 1, 3: 1, 4: 1, 5: 1}
column2计数字典: {'c': 1, 'd': 1, 'e': 1, 'b': 1, 'a': 1}