matplotlibの傾きと切片に基づいて行を追加しますか？

Question

Rには、切片（第1引数）の傾き（第2引数）の指定に基づいてプロット上に線を描くことができるablineという関数があります。例えば、

plot(1:10,1:10) 
abline(0,1) 

0の切片と1の傾斜を有するラインは、プロットの全範囲に及ぶ場合。そのような機能はmatplotlib.pyplotにありますか？

Answer 1

(intercept, slope)が(0, 1)の場合、次の機能を使用して他のスロープとインターセプトに対応することができますが、軸のリミットを変更したり、オートスケールをオンに戻したりすると再調整されません。

def abline(): 
    gca = plt.gca() 
    gca.set_autoscale_on(False) 
    gca.plot(gca.get_xlim(),gca.get_ylim()) 

import matplotlib.pyplot as plt 
plt.scatter(range(10),range(10)) 
abline() 
plt.draw() 

Answer 2

私はコールバックに頼らずにこれを行う方法を見つけられませんでしたが、これはかなりうまくいくようです。

import numpy as np 
from matplotlib import pyplot as plt 


class ABLine2D(plt.Line2D): 

    """ 
    Draw a line based on its slope and y-intercept. Additional arguments are 
    passed to the <matplotlib.lines.Line2D> constructor. 
    """ 

    def __init__(self, slope, intercept, *args, **kwargs): 

     # get current axes if user has not specified them 
     if not 'axes' in kwargs: 
      kwargs.update({'axes':plt.gca()}) 
     ax = kwargs['axes'] 

     # if unspecified, get the current line color from the axes 
     if not ('color' in kwargs or 'c' in kwargs): 
      kwargs.update({'color':ax._get_lines.color_cycle.next()}) 

     # init the line, add it to the axes 
     super(ABLine2D, self).__init__([], [], *args, **kwargs) 
     self._slope = slope 
     self._intercept = intercept 
     ax.add_line(self) 

     # cache the renderer, draw the line for the first time 
     ax.figure.canvas.draw() 
     self._update_lim(None) 

     # connect to axis callbacks 
     self.axes.callbacks.connect('xlim_changed', self._update_lim) 
     self.axes.callbacks.connect('ylim_changed', self._update_lim) 

    def _update_lim(self, event): 
     """ called whenever axis x/y limits change """ 
     x = np.array(self.axes.get_xbound()) 
     y = (self._slope * x) + self._intercept 
     self.set_data(x, y) 
     self.axes.draw_artist(self) 

Answer 3

私はこの質問が数年前であることを知っていますが、受け入れられた答えがないので、私は何が私のために追加されます。

グラフに値をプロットしてから、元のグラフよりも最適な線とプロットの座標の別の値を生成することができます。たとえば、次のコードを参照してください。これらのソリューションの多くは、データをフィットプロットに行を追加することに注力してい

import matplotlib.pyplot as plt 
import numpy as np 

# Some dummy data 
x = [1, 2, 3, 4, 5, 6, 7] 
y = [1, 3, 3, 2, 5, 7, 9] 

# Find the slope and intercept of the best fit line 
slope, intercept = np.polyfit(x, y, 1) 

# Create a list of values in the best fit line 
abline_values = [slope * i + intercept for i in x] 

# Plot the best fit line over the actual values 
plt.plot(x, y, '--') 
plt.plot(x, abline_values, 'b') 
plt.title(slope) 
plt.show() 

Answer 4

X = np.array([1, 2, 3, 4, 5, 6, 7]) 
Y = np.array([1.1,1.9,3.0,4.1,5.2,5.8,7]) 

scatter (X,Y) 
slope, intercept = np.polyfit(X, Y, 1) 
plot(X, X*slope + intercept, 'r') 

Answer 5

。ここでは、傾きと切片に基づいてプロットに任意の線を追加する簡単な方法を示します。

def abline(slope, intercept): 
    """Plot a line from slope and intercept""" 
    axes = plt.gca() 
    x_vals = np.array(axes.get_xlim()) 
    y_vals = intercept + slope * x_vals 
    plt.plot(x_vals, y_vals, '--')