对数据帧中的多个组运行pearson相关性测试

import pandas as pd
import numpy as np
from scipy.stats import pearsonr

data = {'Group': ['A', 'A', 'A', 'B', 'B', 'B'],
        'X': [1, 2, 3, 4, 5, 6],
        'Y': [2, 4, 6, 8, 10, 12]}
df = pd.DataFrame(data)

def calculate_pearson_correlation(group_df):
    correlation, p_value = pearsonr(group_df['X'], group_df['Y'])
    return correlation, p_value

grouped = df.groupby('Group').apply(lambda x: calculate_pear彰son_correlation(x[['X', 'Y']]))

print(grouped)

Group
A    (1.0, 0.0)
B    (1.0, 0.0)
dtype: object