linear_regress.py 文件源码

def Regress(self):
    if len(self._independent_data) == 0:
      print('Loaded no data for contest "%s" party "%s"' % (self._independent, self._party))
      return
    if len(self._dependent_data) == 0:
      print('Loaded no data for contest "%s" party "%s"' % (self._dependent, self._party))
      return
    y = []
    x = []
    for precinct,votes in self._independent_data.iteritems():
      if precinct not in self._dependent_data:
        continue
      x.append(votes)
      y.append(self._dependent_data[precinct])
    if not x or len(x) != len(y):
      print('Mismatched or empty data')
      return
    slope, intercept, r_value, p_value, std_err = stats.linregress(x,y)
    print('Slope=%6.4f Intercept=%6.4f R^2=%.4f p=%.6f' % (slope, intercept,
      r_value**2, p_value))
    self._PrintResiduals(slope, intercept)