Files
Feature-Extraction/correlation_test2.py
2025-10-20 22:01:18 +08:00

45 lines
9.3 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

from scipy.stats import pearsonr
import matplotlib.pyplot as plt
import seaborn as sns
# -------------- 自己填充数据 --------------
RT = [70, 67, 69, 67, 69, 66, 68, 67, 68, 67, 69, 67, 69, 70, 70, 69, 68, 68, 70, 69, 70, 68, 70, 70, 71, 70, 69, 72, 71, 69, 70, 68, 69, 69, 70, 67, 70, 69, 72, 69, 70, 69, 71, 68, 71, 68, 72, 70, 69, 69, 70, 68, 72, 68, 72, 69, 72, 70, 71, 68, 72, 69, 71, 69, 72, 71, 70, 69, 73, 70, 71, 69, 71, 72, 71, 71, 69, 69, 73, 71, 71, 70, 72, 72, 72, 72, 72, 72, 73, 72, 71, 72, 71, 73, 71, 72, 71, 72, 71, 72, 73, 73, 70, 71, 74, 70, 72, 70, 74, 72, 73, 71, 74, 70, 72, 69, 73, 71, 74, 71, 74, 74, 73, 73, 72, 74, 71, 73, 71, 73, 71, 73, 72, 74, 70, 75, 72, 73, 72, 73, 73, 74, 74, 72, 74, 72, 73, 72, 74, 72, 73, 71, 72, 72, 74, 71, 74, 71, 74, 73, 74, 71, 74, 73, 75, 74, 71, 74, 71, 74, 72, 73, 72, 73, 71, 76, 72, 74, 75, 73, 73, 72, 75, 72, 72, 73, 74, 72, 75, 71, 75, 73, 72, 74, 74, 74, 74, 75, 72, 74, 72, 74, 74, 73, 73, 76, 75, 74, 74, 73, 75, 71, 75, 75, 73, 75, 73, 74, 72, 74, 74, 74, 74, 74, 75, 72, 74, 73, 76, 73, 74, 73, 76, 75, 73, 75, 74, 75, 74, 76, 76, 74, 76, 73, 76, 73, 76, 73, 75, 75, 76, 76, 75, 76, 75, 77, 73, 75, 74, 75, 75, 75, 73, 74, 75, 74, 74, 74, 75, 74, 77, 75, 73, 77, 74, 76, 74, 76, 75, 75, 74, 74, 76, 74, 76, 73, 77, 74, 75, 75, 75, 77, 74, 75, 75, 75, 76, 75, 74, 75, 76, 74, 75, 73, 75, 72, 77, 75, 76, 76, 73, 75, 73, 75, 73, 74, 75, 74, 76, 75, 75, 73, 75, 73, 76, 75, 76, 75, 73, 75, 73, 75, 74, 74, 75, 76, 74, 75, 76, 74, 75, 74, 78, 75, 75, 75, 75, 76, 75, 77, 75, 77, 75, 74, 73, 75, 75, 76, 76, 74, 77, 74, 76, 77, 75, 74, 76, 76, 74, 76, 72, 77, 75, 73, 76, 74, 76, 74, 78, 75, 77, 77, 74, 74, 75, 76, 74, 75, 75, 76, 74, 75, 76, 74, 77, 76, 76, 77, 75, 76, 75, 76, 73, 77, 75, 74, 77, 74, 76, 75, 76, 74, 77, 77, 75, 79, 75, 77, 75, 77, 75, 77, 77, 75, 77, 75, 76, 74, 76, 76, 77, 76, 75, 77, 74, 77, 74, 77, 78, 74, 76, 74, 77, 74, 77, 77, 75, 75, 74, 77, 75, 76, 76, 76, 75, 73, 77, 74, 75, 74, 77, 76, 73, 74, 74, 77, 74, 81, 82, 83, 80, 82, 82, 82, 80, 80, 83, 82, 82, 79, 81, 83, 82, 82, 80, 81, 83, 82, 81, 80, 82, 82, 79, 79, 81, 82, 80, 79, 81, 82, 81, 79, 81, 81, 78, 81, 81, 81, 79, 80, 81, 82, 80, 82, 82, 82, 80, 80, 82, 82, 80, 80, 81, 81, 79, 80, 82, 81, 78, 80, 80, 78, 79, 81, 81, 83, 80, 80, 82, 83, 82, 81, 80, 82, 83, 82, 80, 80, 82, 82, 82, 82, 80, 83, 83, 83, 80, 79, 82, 82, 81, 79, 81, 82, 82, 80, 80, 82, 82, 80, 80, 81, 82, 82, 80, 82, 81, 82, 80, 81, 82, 82, 82, 81, 83, 83, 82, 80, 80, 83, 82, 82, 80, 82, 82, 83, 79, 80, 83, 82, 79, 82, 83, 83, 81, 80, 80, 83, 82, 80, 81, 82, 82, 82, 81, 79, 83, 82, 82, 80, 80, 82, 82, 82, 80, 82, 83, 82, 80, 80, 82, 82, 80, 82, 82, 82, 81, 80, 82, 82, 80, 81, 82, 80, 80, 81, 82, 80, 79, 82, 82, 80, 81, 82, 82, 80, 79, 82, 82, 78, 82, 81, 82, 80, 80, 82, 82, 78, 81, 83, 82, 80, 79, 82, 82, 80, 80, 82, 82, 82, 80, 79, 82, 81, 79, 80, 82, 81, 79, 79, 81, 81] # 长度 = n
RT_PAT = [48, 45, 45, 45, 46, 44, 44, 44, 46, 46, 46, 43, 46, 49, 47, 46, 44, 46, 47, 45, 46, 46, 46, 45, 46, 48, 46, 48, 46, 47, 46, 45, 45, 46, 45, 43, 47, 45, 47, 45, 46, 46, 46, 44, 47, 46, 49, 44, 44, 45, 46, 44, 47, 46, 48, 44, 46, 45, 48, 44, 47, 47, 49, 47, 47, 47, 48, 45, 48, 46, 47, 46, 46, 46, 46, 47, 46, 45, 49, 49, 50, 48, 49, 48, 50, 48, 48, 48, 51, 49, 48, 48, 49, 51, 46, 48, 48, 50, 50, 49, 48, 50, 49, 48, 48, 47, 49, 47, 49, 49, 50, 49, 49, 47, 50, 46, 49, 48, 52, 49, 51, 49, 51, 49, 48, 48, 49, 49, 48, 49, 49, 50, 49, 51, 48, 53, 51, 50, 49, 52, 52, 51, 50, 49, 50, 48, 49, 50, 50, 48, 48, 49, 49, 49, 51, 48, 52, 48, 51, 50, 52, 48, 51, 50, 52, 49, 47, 51, 50, 50, 48, 50, 51, 50, 46, 52, 50, 51, 50, 50, 49, 49, 50, 48, 48, 50, 49, 46, 52, 49, 51, 48, 50, 51, 51, 49, 52, 51, 49, 49, 49, 51, 49, 49, 49, 53, 50, 52, 50, 49, 51, 49, 51, 50, 50, 51, 48, 49, 49, 50, 49, 50, 51, 51, 51, 50, 51, 48, 52, 50, 51, 50, 52, 50, 49, 51, 49, 50, 50, 52, 52, 50, 51, 52, 53, 50, 52, 50, 52, 51, 52, 52, 53, 52, 51, 51, 48, 52, 51, 50, 51, 52, 50, 52, 50, 52, 52, 52, 52, 51, 54, 52, 49, 52, 51, 52, 49, 51, 52, 52, 51, 51, 51, 52, 54, 51, 54, 50, 53, 53, 52, 52, 51, 52, 52, 51, 51, 53, 51, 51, 52, 52, 53, 50, 51, 50, 55, 52, 53, 52, 51, 52, 49, 50, 50, 51, 50, 49, 51, 52, 52, 48, 51, 50, 54, 49, 51, 51, 51, 50, 49, 51, 52, 50, 49, 51, 52, 53, 51, 50, 52, 51, 53, 51, 52, 52, 51, 50, 51, 54, 52, 53, 51, 51, 51, 52, 52, 53, 53, 50, 53, 50, 52, 52, 51, 52, 53, 51, 52, 53, 49, 52, 51, 51, 52, 51, 53, 51, 54, 51, 53, 54, 50, 49, 50, 53, 52, 52, 53, 54, 52, 52, 52, 51, 54, 51, 52, 53, 52, 51, 52, 52, 51, 53, 51, 52, 54, 53, 53, 52, 53, 52, 55, 53, 52, 56, 52, 53, 51, 54, 51, 53, 53, 52, 54, 49, 52, 53, 54, 51, 54, 53, 53, 53, 51, 54, 52, 53, 53, 51, 52, 50, 53, 51, 53, 52, 52, 52, 52, 53, 52, 54, 52, 53, 53, 51, 52, 51, 52, 51, 52, 51, 50, 49, 50, 51, 49, 52, 53, 54, 51, 54, 54, 52, 53, 53, 54, 53, 53, 50, 53, 55, 52, 53, 52, 53, 54, 54, 52, 51, 54, 53, 52, 53, 54, 53, 52, 52, 53, 54, 53, 50, 53, 51, 50, 53, 52, 52, 51, 51, 52, 53, 53, 54, 54, 53, 50, 53, 54, 53, 51, 53, 52, 53, 52, 51, 54, 53, 50, 51, 52, 52, 52, 52, 53, 55, 52, 52, 54, 54, 53, 53, 50, 54, 54, 53, 52, 53, 54, 54, 53, 52, 52, 55, 54, 53, 53, 51, 53, 53, 52, 51, 52, 53, 52, 52, 52, 53, 53, 53, 51, 54, 54, 52, 52, 55, 52, 54, 52, 52, 55, 54, 52, 53, 55, 54, 53, 51, 52, 55, 53, 55, 53, 54, 54, 54, 52, 51, 55, 54, 51, 55, 56, 54, 54, 51, 51, 56, 54, 50, 53, 54, 53, 53, 54, 50, 54, 53, 53, 52, 53, 54, 52, 53, 53, 55, 51, 54, 51, 52, 55, 55, 52, 56, 54, 54, 54, 53, 54, 53, 54, 53, 55, 54, 53, 54, 55, 55, 52, 55, 54, 52, 54, 54, 55, 53, 53, 54, 53, 51, 55, 54, 55, 54, 53, 55, 55, 52, 54, 54, 54, 53, 52, 55, 53, 52, 53, 54, 54, 53, 50, 51, 54, 53, 51, 53, 53, 53, 52, 50, 53, 53] # 长度 = n
on_dn = [79, 79, 81, 80, 81, 77, 82, 84, 84, 81, 79, 76, 82, 84, 83, 78, 81, 81, 82, 77, 81, 76, 80, 78, 79, 83, 83, 79, 80, 80, 79, 77, 80, 81, 81, 78, 79, 77, 83, 80, 85, 79, 77, 78, 84, 82, 86, 79, 79, 86, 82, 79, 81, 80, 83, 80, 88, 82, 83, 81, 82, 81, 88, 81, 79, 82, 84, 80, 79, 83, 84, 80, 81, 81, 80, 81, 84, 78, 77, 79, 87, 81, 78, 77, 82, 80, 76, 79, 82, 80, 77, 78, 79, 86, 78, 78, 78, 79, 83, 81, 79, 79, 84, 81, 77, 78, 83, 78, 78, 84, 80, 82, 77, 78, 87, 78, 78, 79, 83, 80, 79, 80, 79, 81, 78, 76, 84, 80, 75, 84, 79, 80, 77, 76, 82, 80, 79, 78, 80, 77, 80, 77, 79, 78, 79, 79, 79, 80, 79, 74, 80, 80, 81, 75, 78, 80, 82, 77, 81, 80, 81, 77, 80, 83, 79, 78, 76, 79, 81, 77, 79, 79, 79, 76, 75, 79, 80, 81, 76, 80, 79, 80, 78, 78, 81, 76, 81, 78, 81, 78, 79, 78, 82, 82, 79, 79, 79, 79, 75, 81, 78, 79, 82, 81, 80, 76, 77, 79, 80, 78, 79, 80, 81, 79, 79, 80, 78, 81, 79, 79, 78, 78, 83, 79, 79, 82, 77, 76, 81, 79, 80, 78, 80, 77, 81, 78, 80, 77, 79, 79, 81, 79, 77, 83, 81, 81, 79, 82, 81, 79, 78, 80, 82, 80, 73, 80, 79, 82, 80, 80, 78, 81, 82, 80, 79, 81, 81, 80, 79, 80, 81, 83, 78, 79, 79, 83, 79, 78, 80, 81, 80, 78, 79, 80, 82, 77, 81, 79, 84, 80, 81, 78, 79, 82, 80, 79, 79, 82, 80, 79, 80, 79, 82, 79, 77, 79, 82, 79, 83, 78, 80, 83, 78, 78, 78, 82, 80, 75, 80, 81, 81, 71, 80, 84, 82, 77, 79, 83, 79, 82, 79, 83, 78, 76, 73, 79, 81, 82, 79, 80, 81, 82, 79, 78, 84, 80, 81, 78, 80, 85, 80, 81, 78, 82, 79, 82, 80, 82, 83, 74, 81, 80, 80, 78, 79, 82, 78, 82, 78, 80, 78, 77, 82, 80, 79, 82, 80, 79, 81, 78, 82, 84, 78, 78, 78, 82, 80, 79, 80, 82, 81, 74, 79, 82, 80, 77, 80, 81, 81, 79, 78, 79, 81, 78, 78, 80, 82, 81, 77, 75, 81, 82, 81, 79, 78, 80, 80, 78, 76, 80, 81, 80, 78, 80, 77, 72, 78, 80, 82, 81, 76, 82, 80, 79, 80, 83, 81, 79, 80, 79, 79, 79, 78, 82, 76, 77, 82, 80, 90, 81, 80, 80, 80, 80, 82, 81, 79, 80, 82, 80, 80, 82, 80, 82, 81, 74, 76, 82, 83, 82, 79, 84, 82, 82, 86, 81, 83, 83, 82, 77, 81, 84, 80, 84, 86, 80, 82, 86, 78, 80, 84, 82, 86, 82, 85, 82, 83, 84, 81, 85, 83, 79, 83, 79, 79, 80, 81, 82, 82, 81, 83, 79, 80, 84, 82, 83, 80, 82, 82, 82, 82, 80, 78, 84, 82, 79, 81, 84, 83, 78, 84, 84, 79, 80, 87, 85, 79, 79, 82, 81, 84, 82, 78, 86, 86, 82, 83, 82, 82, 83, 83, 80, 81, 86, 84, 83, 86, 80, 81, 81, 81, 81, 84, 84, 83, 81, 89, 83, 84, 82, 82, 84, 83, 81, 79, 83, 80, 83, 80, 80, 83, 83, 81, 83, 81, 84, 80, 81, 82, 83, 82, 84, 82, 81, 85, 82, 83, 78, 82, 83, 79, 83, 85, 82, 84, 80, 81, 85, 81, 79, 83, 84, 82, 81, 85, 79, 81, 84, 84, 82, 82, 85, 81, 80, 83, 84, 79, 87, 82, 82, 87, 85, 80, 85, 83, 81, 83, 81, 83, 84, 83, 82, 85, 83, 81, 84, 84, 84, 79, 85, 87, 81, 78, 83, 84, 82, 80, 80, 81, 81, 83, 81, 85, 84, 80, 85, 83, 82, 82, 81, 83, 82, 81, 83, 82, 80, 81, 83, 86, 82, 80, 80, 84, 83, 80, 83, 79, 84, 82, 79, 82, 85] # 长度 = n
# -----------------------------------------
n = len(on_dn)
# 1. 计算
r1, p1 = pearsonr(RT, on_dn)
r2, p2 = pearsonr(RT_PAT, on_dn)
# 2. 画图
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(5, 8))
# 图1RT vs on-dn
sns.regplot(x=RT, y=on_dn,
scatter_kws={'alpha': 0.7, 'color': 'dodgerblue'},
line_kws={'color': 'red',
'label': rf'r={r1:.3f}, n={n}, p={p1:.3g}'},
ax=ax1)
ax1.set_xlabel('RT')
ax1.set_ylabel('on-dn')
ax1.set_title(rf'$\it r$={r1:.3f}, $\it p$={p1:.3g}, $\it n$={n}')
ax1.legend()
ax1.grid(True, linestyle='--', alpha=0.4)
# 图2RT-PAT vs on-dn
sns.regplot(x=RT_PAT, y=on_dn,
scatter_kws={'alpha': 0.7, 'color': 'dodgerblue'},
line_kws={'color': 'red',
'label': rf'r={r2:.3f}, n={n}, p={p2:.3g}'},
ax=ax2)
ax2.set_xlabel('RT-PAT')
ax2.set_ylabel('on-dn')
ax2.set_title(rf'$\it r$={r2:.3f}, $\it p$={p2:.3g}, $\it n$={n}')
ax2.legend()
ax2.grid(True, linestyle='--', alpha=0.4)
plt.tight_layout()
plt.show()