@@ -750,7 +750,7 @@ def update_data(self, attr_x, attr_y, reset_view=True):
750750
751751 def draw_regression_line (self , x_data , y_data , min_x , max_x ):
752752 if self .show_reg_line and self .can_draw_regresssion_line ():
753- slope , intercept , _ , _ , _ = linregress (x_data , y_data )
753+ slope , intercept , rvalue , _ , _ = linregress (x_data , y_data )
754754 start_y = min_x * slope + intercept
755755 end_y = max_x * slope + intercept
756756 angle = np .degrees (np .arctan ((end_y - start_y ) / (max_x - min_x )))
@@ -760,7 +760,7 @@ def draw_regression_line(self, x_data, y_data, min_x, max_x):
760760 rotateAxis = (1 , 0 ), movable = True )
761761 self .reg_line_item = InfiniteLine (
762762 pos = QPointF (min_x , start_y ), pen = pg .mkPen (color = color , width = 1 ),
763- angle = angle , label = "r = {:.2f}" .format (slope ), labelOpts = l_opts )
763+ angle = angle , label = "r = {:.2f}" .format (rvalue ), labelOpts = l_opts )
764764 if rotate :
765765 self .reg_line_item .label .angle = 180
766766 self .reg_line_item .label .updateTransform ()
0 commit comments