1. ホーム
  2. python

[解決済み] Matplotlib で傾きと切片に基づく直線を追加するには?

2022-02-16 19:11:10

質問

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

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

ここで、切片が 0 で傾きが 1 の線は、プロットの全範囲に及びます。Matplotlib にそのような関数はありますか?

どのように解決するのですか?

これらの解決策の多くは、データに適合する直線をプロットに追加することに焦点を当てています。ここでは、傾きと切片に基づいて任意の直線をプロットに追加するための簡単な解決策を紹介します。

import matplotlib.pyplot as plt 
import numpy as np    

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, '--')