针对不同策略的股票池打架如何处理

import pandas as pd
def analyze_strategy_overlap(strategy_dfs):
    """
    综合分析多个策略的选股重叠情况
    参数:
    strategy_dfs: 包含多个策略选股DataFrame的列表
    返回:
    包含个股出现次数和详细策略信息的DataFrame
    """
    # 收集所有股票的代码和名称
    all_stocks = pd.DataFrame()
    for i, df in enumerate(strategy_dfs):
        temp_df = df[['股票代码', '股票名称']].copy()
        temp_df['策略来源'] = f'策略{i + 1}'
        all_stocks = pd.concat([all_stocks, temp_df], ignore_index=True)
    # 统计每只股票的出现次数
    count_summary = all_stocks['股票代码'].value_counts().reset_index()
    count_summary.columns = ['股票代码', '出现次数']
    # 添加股票名称
    stock_name_map = all_stocks[['股票代码', '股票名称']].drop_duplicates()
    stock_name_map = stock_name_map.set_index('股票代码')['股票名称'].to_dict()
    count_summary['股票名称'] = count_summary['股票代码'].map(stock_name_map)
    # 收集每个股票出现在哪些策略中
    strategy_occurrence = {}
    for stock in count_summary['股票代码']:
        strategies = all_stocks[all_stocks['股票代码'] == stock]['策略来源'].unique()
        strategy_occurrence[stock] = ', '.join(strategies)
    count_summary['出现策略'] = count_summary['股票代码'].map(strategy_occurrence)
    # 按出现次数降序排序
    count_summary = count_summary.sort_values('出现次数', ascending=False)
    return count_summary
# 使用示例
# 假设我们有三个策略的选股结果
strategy1 = pd.DataFrame({
    '股票代码': ['000001.SZ', '000002.SZ', '600036.SH', '601318.SH', '000858.SZ'],
    '股票名称': ['平安银行', '万科A', '招商银行', '中国平安', '五粮液']
})
strategy2 = pd.DataFrame({
    '股票代码': ['000001.SZ', '600036.SH', '601318.SH', '600519.SH', '000651.SZ'],
    '股票名称': ['平安银行', '招商银行', '中国平安', '贵州茅台', '格力电器']
})
strategy3 = pd.DataFrame({
    '股票代码': ['000001.SZ', '601318.SH', '600519.SH', '000333.SZ', '002415.SZ'],
    '股票名称': ['平安银行', '中国平安', '贵州茅台', '美的集团', '海康威视']
})
# 分析策略重叠
result = analyze_strategy_overlap([strategy1, strategy2, strategy3])
print("多策略综合分析结果:")
print(result)