我有一点组合问题,我甚至难以概念化如何使用itertools来解决它。我的数据集如下所示:
GoldMeasure Measure1 Measure2 Measure3 Measure4 Measure5 3+4-5
30.501 -1 -1 -1 -1 -1 -1
30.658 -1 -1 0 -1 -1 0
31.281 -1 -1 -1 1 -3
31.506 7 -1 -1 1 -1 1
31.554 -1 -1 -1 -1 -1 -1
31.613 -1 -1 -1 1 0
31.838 -1 -1 -1 -1 -2
31.954 -1 -1 -1 1 -1 1
33.073 1 1 1 -1 2
33.592 -2 -2 2 0 -2 4
Coefficient: -0.054 0.119 0.690 0.474 -0.441 0.723
我正在尝试找到具有最高Pearson相关系数的度量1-5与Gold Measure的最佳组合。在上面的数据集中,最佳单一度量是度量三,系数为0.69。但是,如果将“测量三”添加到“测量四”然后减去“测量五”,则会得到一个系数为0.72的新组合度量。基本上,我想确定这五个指标的所有可能组合的相关系数。这些组合可以是2/5的措施,3/5的措施(如上例中所示),4/5的措施或所有五个措施。
订单有点重要。确实,措施3,4和5的添加剂组合与5,3和4的添加剂组合相同;但是,我也试图将算术运算符作为另一个选项(加法或减法)。因此,3 + 4 + 5与5 + 3 + 4相同,但3 + 4-5与5 + 3-4不同。
非常感谢任何指导或帮助。实际数据集有46个度量,并且,如上面的示例数据集中,有一些度量没有特定样本的值。这些措施的价值既是消极的,也是积极的,但不是有限的。
提前感谢您的帮助。
答案 0 :(得分:1)
采用五项措施做到这一点是可行的。
from itertools import product
from numpy import negative, array_equal
choices = [1, 0, -1]
measures = 5
limit = len(choices)**measures//2+1
count = 0
measure_combinations = []
for p in product(*([choices]*5)):
measure_combinations.append(list(p))
count += 1
if count == limit:
break
print (len(measure_combinations))
例如,measure_combinations之一的内积,比如[1,1,0,-1,-1]与度量列的每一行将提供一系列值,GoldMeasure可以在这些值上进行回归以获得相关系数,这可以用于122种独特的可能性。
然而,对于41项措施,将有18,236,498,188,585,393,202个独特的可能性,基于41次重复{-1,0,1}的部分产品。
答案 1 :(得分:1)
from collections import defaultdict
import statistics
import itertools
import scipy.stats
import heapq
#Please add a filename here and use a .csv format.
F=open('filename','r').readlines();
F[0].strip().split(',');
HEADER=[r.strip() for r in F[0].strip().split(',')[2:]];
DDF={};
DDP={};
DDN={};
DDG={};
#Please replace missing values using Excel into "NAA". I have replaced the "NAA"/missing values with average of the column. You may also consider ignoring missing rows before doing correlation between 2 columns.
for f in F[1:]:
DATA=f.strip().split(',');
ATAD=[[i.strip(),j.strip()] for i,j in zip(HEADER,DATA[2:])];
for atad in ATAD:
if atad[0].strip() in DDF.keys():
DDF[atad[0].strip()].append(atad[1].strip());
else:
DDF[atad[0].strip()]=[atad[1].strip()];
if F[0].strip().split(',')[1].strip() in DDG.keys():
DDG[F[0].strip().split(',')[1].strip()].append(float(DATA[1].strip()));
else:
DDG[F[0].strip().split(',')[1].strip()]=[float(DATA[1].strip())];
for ke in DDF.keys():
AVGP=statistics.mean([float(u) for u in DDF[ke.strip()] if u.strip()!='NAA']);
NEWP=[float(nr.strip()) if nr.strip()!='NAA' else AVGP for nr in DDF[ke.strip()]];
if ke in DDP.keys():
DDP[ke.strip()]=NEWP;
else:
DDP[ke.strip()]=NEWP;
AVGN=statistics.mean([-1*float(e) for e in DDF[ke.strip()] if e.strip()!='NAA']);
NEWN=[-1*float(ne.strip()) if ne.strip()!='NAA' else AVGN for ne in DDF[ke.strip()]];
if 'minus_'+ke.strip() in DDN.keys():
DDN['minus_'+ke.strip()]=NEWN;
else:
DDN['minus_'+ke.strip()]=NEWN;
U=0;
L1=DDP.keys()+DDN.keys();
N=range(1,len(L1));
U=[0,0,-1];
for n in N:
DDU=[];
for subset in itertools.combinations(L1,n):
S=[];
SSET=[sset.strip().replace('minus_','') if sset.strip().startswith('minus_') else sset.strip() for sset in list(subset)];
if len(set(SSET))>=len(subset):
TMP=[DDN[p.strip()] if p.strip().startswith('minus_') else DDP[p.strip()] for p in subset];
for y in range(0,len(TMP[0])):
for x in TMP:
S.append(x[y]);
SUM=[sum(S[w:w + n]) for w in range(0, len(S),n)];
K=['+'.join(list(subset)).strip()]+[' vs Cq TREC']+list(scipy.stats.pearsonr(SUM,DDG['Cq TREC']));
if str(K[-2])!='nan':
DDU.append([K[-2],K]);
DDU.sort(key=lambda x: x[0],reverse=True);
for ea in DDU[0:3]:
print repr(n).strip()+','+ea[1][0].strip().replace('+minus_','-').replace('minus_','-')+' '+ea[1][1].strip()+','+repr(ea[1][-2]).strip();