web-dev-qa-db-ja.com

Matplotlibで与えられたプロットに垂直線を引く方法は？

時間表示で信号をプロットした場合、対応する時間インデックスをマークする線をどのように引くか？

具体的には、0から2.6（s）の範囲のタイムインデックスを持つ信号プロットで、リスト[0.22058956, 0.33088437, 2.20589566]に対応するタイムインデックスを示す垂直の赤い線を描きたいのですが、どうすればよいですか

pythonmatplotlib
188
Francis

実際の高さを指定しなくても、プロットウィンドウ全体をカバーする垂直線を追加する標準的な方法はplt.axvlineです。

import matplotlib.pyplot as plt

plt.axvline(x=0.22058956)
plt.axvline(x=0.33088437)
plt.axvline(x=2.20589566)

OR

xcoords = [0.22058956, 0.33088437, 2.20589566]
for xc in xcoords:
    plt.axvline(x=xc)

他のプロットコマンドで利用可能な多くのキーワードを使用することができます（例：color、linestyle、linewidth ...）。 Axes座標で好きな場合は、キーワード引数yminとymaxを渡すことができます（例えばymin=0.25、ymax=0.75はプロットの中央半分をカバーします）。水平線（axhline）と長方形（axvspan）に対応する関数があります。

314
Gabriel

複数行の場合

xposition = [0.3, 0.4, 0.45]
for xc in xposition:
    plt.axvline(x=xc, color='k', linestyle='--')
37
Qina Yan

他の人が示唆しているように、ループ内でaxvlineを呼び出すことは機能しますが、不便な場合があります。

  1. 各行は別々のプロットオブジェクトであるため、多数の行があると処理が非常に遅くなります。
  2. 凡例を作成すると、各行に新しいエントリが追加されますが、これはあなたが望むものではないかもしれません。

代わりに、すべての線を単一のプロットオブジェクトとして作成する次の便利な機能を使用できます。

import matplotlib.pyplot as plt
import numpy as np


def axhlines(ys, ax=None, **plot_kwargs):
    """
    Draw horizontal lines across plot
    :param ys: A scalar, list, or 1D array of vertical offsets
    :param ax: The axis (or none to use gca)
    :param plot_kwargs: Keyword arguments to be passed to plot
    :return: The plot object corresponding to the lines.
    """
    if ax is None:
        ax = plt.gca()
    ys = np.array((ys, ) if np.isscalar(ys) else ys, copy=False)
    lims = ax.get_xlim()
    y_points = np.repeat(ys[:, None], repeats=3, axis=1).flatten()
    x_points = np.repeat(np.array(lims + (np.nan, ))[None, :], repeats=len(ys), axis=0).flatten()
    plot = ax.plot(x_points, y_points, scalex = False, **plot_kwargs)
    return plot


def axvlines(xs, ax=None, **plot_kwargs):
    """
    Draw vertical lines on plot
    :param xs: A scalar, list, or 1D array of horizontal offsets
    :param ax: The axis (or none to use gca)
    :param plot_kwargs: Keyword arguments to be passed to plot
    :return: The plot object corresponding to the lines.
    """
    if ax is None:
        ax = plt.gca()
    xs = np.array((xs, ) if np.isscalar(xs) else xs, copy=False)
    lims = ax.get_ylim()
    x_points = np.repeat(xs[:, None], repeats=3, axis=1).flatten()
    y_points = np.repeat(np.array(lims + (np.nan, ))[None, :], repeats=len(xs), axis=0).flatten()
    plot = ax.plot(x_points, y_points, scaley = False, **plot_kwargs)
    return plot
19
Peter

誰かがいくつかの垂直線にlegendやcolorsを追加したいのなら、これを使います：

import matplotlib.pyplot as plt

xcoords = [0.1, 0.3, 0.5]
colors = ['r','k','b']

for xc,c in Zip(xcoords,colors):
    plt.axvline(x=xc, label='line at x = {}'.format(xc), c=c)

plt.legend()
plt.show()

結果：

my amazing plot seralouk

9
makaros

上記の回答で提供されているplt.axvlineおよびplt.plot((x1, x2), (y1, y2)) OR plt.plot([x1, x2], [y1, y2])に加えて、次のものも使用できます。

plt.vlines(x_pos, ymin=y1, ymax=y2)

x_posからy1までの範囲でy2に垂直線をプロットします。ここで、値y1とy2は絶対データ座標です。

8
Sheldore