2次元で配置されたAxesを一括で扱う

2020-03-08 / tau / コメントする

subplots()やadd_subplot()で複数の行数・列数のAxesを生成すると、Axesオブジェクトの2次元の配列となる。

この結果に対して一律に処理をしたい場合（たとえば軸の値や凡例を設定したい、アスペクトを揃えたいなどの場合）、いちいち二重ループを回すのが面倒。

import numpy as np
import matplotlib.pyplot as plt

fig, axs = plt.subplots(2, 2)

x = np.linspace(-np.pi, np.pi)
n = 1
for row in axs:
    for ax in row:
        ax.plot(x, np.sin(n*x), label="n={}".format(n))
        n += 1
        ax.set_ylim(-1.2, 1.8)
        ax.legend(loc='upper left')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

fig, axs = plt.subplots(2, 2)

x = np.linspace(-np.pi, np.pi)

n = 1

for row in axs:

for ax in row:

ax.plot(x, np.sin(n*x), label="n={}".format(n))

n += 1

ax.set_ylim(-1.2, 1.8)

ax.legend(loc='upper left')

plt.show()

これを1次元配列に変換して一括で扱う方法。

変換方法の1つは、以下のように1次元配列で取り出してしまう方法

axs_1d = [ax for row in axs for ax in row]

1	axs_1d = [ax for row in axs for ax in row]

あるいは、以下のように2次元配列を1次元に変換する方法（当初、reshape(1, -1)[0]のようなことをしていたが、reshape(-1)とすればよいことがわかった）。

axs_1d = axs.reshape(-1)

1	axs_1d = axs.reshape(-1)

こうすると1次元配列axs_1dで2次元のaxsの全要素に対してアクセス可能になる。

import numpy as np
import matplotlib.pyplot as plt

fig, axs = plt.subplots(2, 2)

axs_1d = axs.reshape(-1)

x = np.linspace(-np.pi, np.pi)
for n, ax in enumerate(axs_1d):
    ax.plot(x, np.sin(n*x), label="n={}".format(n))
    ax.set_ylim(-1.2, 1.8)
    ax.legend(loc='upper left')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

fig, axs = plt.subplots(2, 2)

axs_1d = axs.reshape(-1)

x = np.linspace(-np.pi, np.pi)

for n, ax in enumerate(axs_1d):

ax.plot(x, np.sin(n*x), label="n={}".format(n))

ax.set_ylim(-1.2, 1.8)

ax.legend(loc='upper left')

plt.show()

このほかにflatten()、ravel()を使う方法もある。flatten()はコピーを返すが、Axesオブジェクトへの参照先は変わらないので同じ効果。

各グラフにカウンターの値を適用するときはenumerate、他のリストなどと同時に変えていくときはzipを使う。

matplotlib.pyplot.hist – ヒストグラム

2020-03-08 / tau / コメントする

概要

matplotlib.pyplot.histは、配列データのヒストグラムを描画する。主なパラメータのみ示す。

hist(X, bins, range, density, cumulative, histtype, rwidth, color, stacked): Xはヒストグラムのデータで一つのグラフに一つの1次元配列。その他のパラメーターは以下で説明。

ヒストグラムの形式

度数分布

1次元の配列でデータを渡すと、その度数分布が描かれる。

ただしデフォルトでは各ビンのエッジが区別できないため、これを描くためにはedghecolorの指定が必要。edgecolorの代わりにecで指定してもよい。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x)
axs[1].hist(x, edgecolor='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x)

axs[1].hist(x, edgecolor='k')

plt.show()

頻度分布

density=Trueを指定すると、頻度分布になる。各ビンの総和が1となるように調整され、形状は度数分布と同じ。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x, ec='k')
axs[1].hist(x, density=True, ec='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x, ec='k')

axs[1].hist(x, density=True, ec='k')

plt.show()

累積分布図

cumulative=Trueを指定すると、累積度数分布、累積頻度分布を描く。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x, cumulative=True, ec='k')
axs[1].hist(x, cumulative=True, density=True, ec='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x, cumulative=True, ec='k')

axs[1].hist(x, cumulative=True, density=True, ec='k')

plt.show()

ビン数

binsでヒストグラムの柱（ビン）の数等を指定する。デフォルトはbins=10。

bins=n: ビンの数を数値で指定する。
bins=sequence: ビンの境界値をリスト等で指定する。

ビン(bin)は英語で、店の商品や工場の部品などを入れておく大きなケース、ストッカーのことをいい、British Englishではごみ箱を指す。日本語の瓶(びん)の呼び名とは関係ないらしい。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 2, figsize=(6.4, 2.4))

axs[0].hist(x, bins=5, ec='k')
axs[1].hist(x, bins=[-3, -2, -1, 0, 0.5, 1, 1.5, 2], ec='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 2, figsize=(6.4, 2.4))

axs[0].hist(x, bins=5, ec='k')

axs[1].hist(x, bins=[-3, -2, -1, 0, 0.5, 1, 1.5, 2], ec='k')

plt.show()

レンジ

rangeでヒストグラムのビンを分割する範囲を指定する。そのレンジの上下限値の間でbinsで指定されたビンに分割される。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 3, figsize=(9.6, 2.4))

axs[0].hist(x, edgecolor='k')
axs[1].hist(x, range=(-3, 3), ec='k')
axs[2].hist(x, range=(-1.5, 1.5), ec='k')

for ax in axs:
    ax.set_xlim(-3, 3)
    ax.set_xticks(np.arange(-3, 4, 1))

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 3, figsize=(9.6, 2.4))

axs[0].hist(x, edgecolor='k')

axs[1].hist(x, range=(-3, 3), ec='k')

axs[2].hist(x, range=(-1.5, 1.5), ec='k')

for ax in axs:

ax.set_xlim(-3, 3)

ax.set_xticks(np.arange(-3, 4, 1))

plt.show()

色・エッジの指定

colorでビンの色、edgecolor/ecでエッジの色、linewidthでエッジの幅を指定する。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 2, figsize=(6.4, 2.4))

axs[0].hist(x, color='lightseagreen', edgecolor='k')
axs[1].hist(x, color='lightcoral', ec='navy', linewidth=3)

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 2, figsize=(6.4, 2.4))

axs[0].hist(x, color='lightseagreen', edgecolor='k')

axs[1].hist(x, color='lightcoral', ec='navy', linewidth=3)

plt.show()

ビンの幅

rwidthでビンの幅を指定する。デフォルトはrwidth=1で各ビンが密着。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 2, figsize=(6.4, 2.4))

axs[0].hist(x, rwidth=0.8, ec='k')
axs[1].hist(x, bins=13, width=1.2, ec='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 2, figsize=(6.4, 2.4))

axs[0].hist(x, rwidth=0.8, ec='k')

axs[1].hist(x, bins=13, width=1.2, ec='k')

plt.show()

ヒストグラムのタイプ

histtype='type'でヒストグラムのタイプを指定する。

bar: 一般的なヒストグラムの形状。
barstacked: 複数のヒストグラムの場合に、同じビンの値を積み上げていく。
step: ビンの間の境界を描かない。
stepfilled: ビンの間の境界を描かず、中を塗りつぶす。

以下の例では、'bar'、'step'、'stepfilled'について例示。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 3, figsize=(9.6, 2.4))

axs[0].hist(x, histtype='bar', ec='k')
axs[1].hist(x, histtype='step', ec='k')
axs[2].hist(x, histtype='stepfilled', ec='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, axs = plt.subplots(1, 3, figsize=(9.6, 2.4))

axs[0].hist(x, histtype='bar', ec='k')

axs[1].hist(x, histtype='step', ec='k')

axs[2].hist(x, histtype='stepfilled', ec='k')

plt.show()

複数のヒストグラム

単純な重ね合わせ

複数のヒストグラムを重ね合わせるには、同じターゲットに対して各データについてhistを実行する。

単に重ね合わせると、初めの方のヒストグラムが後の方で塗りつぶされてしまうので、それらを見えるようにするにはalphaで透明度を指定する。

ただし単に重ね合わせただけの場合、各ヒストグラムのビンの境界が必ずしも一致しない。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)
x2 = np.random.normal(0, 1, 500)
x3 = np.random.normal(2, 1, 500)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x1, color='c', edgecolor='k')
axs[0].hist(x2, color='m', edgecolor='k')
axs[0].hist(x3, color='g', edgecolor='k')

axs[1].hist(x1, color='c', edgecolor='k', alpha=0.5)
axs[1].hist(x2, color='m', edgecolor='k', alpha=0.5)
axs[1].hist(x3, color='g', edgecolor='k', alpha=0.5)

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)

x2 = np.random.normal(0, 1, 500)

x3 = np.random.normal(2, 1, 500)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x1, color='c', edgecolor='k')

axs[0].hist(x2, color='m', edgecolor='k')

axs[0].hist(x3, color='g', edgecolor='k')

axs[1].hist(x1, color='c', edgecolor='k', alpha=0.5)

axs[1].hist(x2, color='m', edgecolor='k', alpha=0.5)

axs[1].hist(x3, color='g', edgecolor='k', alpha=0.5)

plt.show()

ヒストグラムとplotの重ね合わせ

こちらを参照。

ビン境界の整合

複数のヒストグラムのビンの境界を一致させるには以下の方法がある。

各グラフに対して同じrangeとbins=nを指定する
各グラフに対して同じbins=sequenceを指定する

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)
x2 = np.random.normal(0, 1, 500)
x3 = np.random.normal(2, 1, 500)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x1, range=(-4, 4), bins=20, color='c', ec='k', alpha=0.5)
axs[0].hist(x2, range=(-4, 4), bins=20, color='m', ec='k', alpha=0.5)
axs[0].hist(x3, range=(-4, 4), bins=20, color='g', ec='k', alpha=0.5)

bn = np.linspace(-4, 4, 21)
axs[1].hist(x1, bins=bn, color='c', ec='k', alpha=0.5)
axs[1].hist(x2, bins=bn, color='m', ec='k', alpha=0.5)
axs[1].hist(x3, bins=bn, color='g', ec='k', alpha=0.5)

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)

x2 = np.random.normal(0, 1, 500)

x3 = np.random.normal(2, 1, 500)

fig, axs = plt.subplots(1,2, figsize=(6.4, 2.4))

axs[0].hist(x1, range=(-4, 4), bins=20, color='c', ec='k', alpha=0.5)

axs[0].hist(x2, range=(-4, 4), bins=20, color='m', ec='k', alpha=0.5)

axs[0].hist(x3, range=(-4, 4), bins=20, color='g', ec='k', alpha=0.5)

bn = np.linspace(-4, 4, 21)

axs[1].hist(x1, bins=bn, color='c', ec='k', alpha=0.5)

axs[1].hist(x2, bins=bn, color='m', ec='k', alpha=0.5)

axs[1].hist(x3, bins=bn, color='g', ec='k', alpha=0.5)

plt.show()

並べる・積み上げる

複数のデータを配列とした場合（複数の1次元のデータを並べて2次元配列として与えた場合）、デフォルトでは各ビンが横に並べられる。

また、stacked=Trueあるいはhisttype='barstacked'を指定した場合、同じ階級のビンが積み上げられる。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)
x2 = np.random.normal(0, 1, 500)
x3 = np.random.normal(2, 1, 500)
X = [x1, x2, x3]

fig, axs = plt.subplots(1,3, figsize=(9.6, 2.4))

axs[0].hist(X, bins=20, ec='k')
axs[1].hist(X, bins=20, stacked=True, ec='k')
axs[2].hist(X, bins=20, histtype='barstacked', ec='k')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)

x2 = np.random.normal(0, 1, 500)

x3 = np.random.normal(2, 1, 500)

X = [x1, x2, x3]

fig, axs = plt.subplots(1,3, figsize=(9.6, 2.4))

axs[0].hist(X, bins=20, ec='k')

axs[1].hist(X, bins=20, stacked=True, ec='k')

axs[2].hist(X, bins=20, histtype='barstacked', ec='k')

plt.show()

戻り値

n: 各ビンの値(度数または頻度)
bins: 各ビンの境界値
patches: ヒストグラム描画に使われたpatcheオブジェクトのリスト

単一のヒストグラムの場合

以下の例では、nの結果として10個のビンの度数が得られている。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, ax = plt.subplots(1, 1, figsize=(6.4, 2.4))

n, bins, patches = ax.hist(x)

print(n)
print(bins)
print(patches)

plt.show()

# [0.02073508 0.10367541 0.14514557 0.26955606 0.35249639 0.37323147
#  0.33176131 0.2280859  0.14514557 0.10367541]
# [-2.55298982 -2.07071537 -1.58844093 -1.10616648 -0.62389204 -0.1416176
#   0.34065685  0.82293129  1.30520574  1.78748018  2.26975462]
# <a list of 10 Patch objects>

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x = np.random.normal(0, 1, 100)

fig, ax = plt.subplots(1, 1, figsize=(6.4, 2.4))

n, bins, patches = ax.hist(x)

print(n)

print(bins)

print(patches)

plt.show()

# [0.02073508 0.10367541 0.14514557 0.26955606 0.35249639 0.37323147

# 0.33176131 0.2280859 0.14514557 0.10367541]

# [-2.55298982 -2.07071537 -1.58844093 -1.10616648 -0.62389204 -0.1416176

# 0.34065685 0.82293129 1.30520574 1.78748018 2.26975462]

# <a list of 10 Patch objects>

複数のヒストグラムの配列の場合

複数のデータを配列で与えた場合の戻り値。nが3つのデータごとの配列になっている。

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)
x2 = np.random.normal(0, 1, 500)
x3 = np.random.normal(2, 1, 500)
X = [x1, x2, x3]

fig, ax = plt.subplots(1,1, figsize=(3.2, 2.4))

n, bins, patches = ax.hist(X, bins=20, ec='k')

print(n)
print(bins)
print(patches)

plt.show()

# [array([ 6., 10., 34., 59., 98., 92., 87., 62., 24., 20.,  8.,  0.,  0.,
#         0.,  0.,  0.,  0.,  0.,  0.,  0.]), array([ 0.,  0.,  0.,  2.,  5., 12., 33., 64., 89., 99., 89., 55., 33.,
#        12.,  6.,  1.,  0.,  0.,  0.,  0.]), array([  0.,   0.,   0.,   0.,   0.,   0.,   0.,   2.,   2.,  11.,  27.,
#         54.,  89.,  89., 106.,  66.,  34.,  12.,   6.,   2.])]
# [-4.77259276 -4.27541438 -3.778236   -3.28105763 -2.78387925 -2.28670087
#  -1.7895225  -1.29234412 -0.79516574 -0.29798737  0.19919101  0.69636938
#   1.19354776  1.69072614  2.18790451  2.68508289  3.18226127  3.67943964
#   4.17661802  4.6737964   5.17097477]
# <a list of 3 Lists of Patches objects>

import numpy as np

import matplotlib.pyplot as plt

np.random.seed(0)

x1 = np.random.normal(-2, 1, 500)

x2 = np.random.normal(0, 1, 500)

x3 = np.random.normal(2, 1, 500)

X = [x1, x2, x3]

fig, ax = plt.subplots(1,1, figsize=(3.2, 2.4))

n, bins, patches = ax.hist(X, bins=20, ec='k')

print(n)

print(bins)

print(patches)

plt.show()

# [array([ 6., 10., 34., 59., 98., 92., 87., 62., 24., 20., 8., 0., 0.,

# 0., 0., 0., 0., 0., 0., 0.]), array([ 0., 0., 0., 2., 5., 12., 33., 64., 89., 99., 89., 55., 33.,

# 12., 6., 1., 0., 0., 0., 0.]), array([ 0., 0., 0., 0., 0., 0., 0., 2., 2., 11., 27.,

# 54., 89., 89., 106., 66., 34., 12., 6., 2.])]

# [-4.77259276 -4.27541438 -3.778236 -3.28105763 -2.78387925 -2.28670087

# -1.7895225 -1.29234412 -0.79516574 -0.29798737 0.19919101 0.69636938

# 1.19354776 1.69072614 2.18790451 2.68508289 3.18226127 3.67943964

# 4.17661802 4.6737964 5.17097477]

# <a list of 3 Lists of Patches objects>

matplotlib.pyplot.quiver – ベクトル場

2020-02-23 / tau / 3件のコメント

概要

matplotlib.pyplot.quiver()はベクトル場を可視化する。基本的なパラメーターは以下の通り。

quiver(X, Y, U, V, [C]): X, Yはベクトルの開始点、U, Vはベクトルの成分、Cはベクトルの大きさに応じたカラーマップ上の色をつけるための配列。

単一のベクトルの描画例

以下の例では、始点の位置と成分を1つずつ指定してベクトルを描画している。デフォルトではベクトルのスケールは描画領域に対して自動的に調節されるが、ここでは描画領域のスケールと同じになるようパラメーターを設定している。

matplotlibのドキュメントでは、scale_unitsのところに以下のように書かれている。

“To plot vectors in the x-y plane, with u and v having the same units as x and y, use angles='xy', scale_units='xy', scale=1”

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

ax.quiver(1, 2, color='blue', angles='xy', scale_units='xy', scale=1)
ax.quiver(1, 2, 2, -1, color='red', angles='xy', scale_units='xy', scale=1)

ax.set_xlim(0, 3)
ax.set_ylim(0, 3)

ax.set_xticks(np.arange(0, 4, 1))
ax.set_yticks(np.arange(0, 4, 1))

ax.grid()
ax.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

fig, ax = plt.subplots()

ax.quiver(1, 2, color='blue', angles='xy', scale_units='xy', scale=1)

ax.quiver(1, 2, 2, -1, color='red', angles='xy', scale_units='xy', scale=1)

ax.set_xlim(0, 3)

ax.set_ylim(0, 3)

ax.set_xticks(np.arange(0, 4, 1))

ax.set_yticks(np.arange(0, 4, 1))

ax.grid()

ax.set_aspect('equal')

plt.show()

ベクトル場の描画例

以下の例では、xy平面上の位置に応じた成分を持つベクトルを描画している。関数のgradientのイメージ。

描画にあたって、開始点の座標とベクトルの成分をmeshgridで生成している。

1つ目の図は単に開始点と成分を与えただけで、単一の色で、ベクトルのスケールは自動調節されている。

2つ目の図はスケールと色付けのための配列を指定し、ベクトルの大きさに応じてcolormapで色を付けている。

import numpy as np
import matplotlib.pyplot as plt

x = y = np.arange(-5, 6)
u = 2 * x
v = 3 * y

x, y = np.meshgrid(x, y)
u, v = np.meshgrid(u, v)

fig, axes = plt.subplots(1, 2, figsize=(12, 4.8))

lim = 8
for ax in axes:
    ax.set_xlim(-lim, lim)
    ax.set_ylim(-lim, lim)

    ax.set_xticks(np.arange(-lim, lim, 1))
    ax.set_yticks(np.arange(-lim, lim, 1))

    ax.grid()
    ax.set_aspect('equal')

C = np.sqrt(u * u + v * v)
axes[0].quiver(x, y, u, v)
axes[1].quiver(x, y, u, v, C, scale=100, cmap='Blues')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = y = np.arange(-5, 6)

u = 2 * x

v = 3 * y

x, y = np.meshgrid(x, y)

u, v = np.meshgrid(u, v)

fig, axes = plt.subplots(1, 2, figsize=(12, 4.8))

lim = 8

for ax in axes:

ax.set_xlim(-lim, lim)

ax.set_ylim(-lim, lim)

ax.set_xticks(np.arange(-lim, lim, 1))

ax.set_yticks(np.arange(-lim, lim, 1))

ax.grid()

ax.set_aspect('equal')

C = np.sqrt(u * u + v * v)

axes[0].quiver(x, y, u, v)

axes[1].quiver(x, y, u, v, C, scale=100, cmap='Blues')

plt.show()

Axes.spines～軸の設定

2020-02-22 / tau / コメントする

概要

グラフのx軸、y軸の位置や表示の有無については、Axesオブジェクトのspinesプロパティーで制御する。

spinesは辞書型でbottom、top、left、rightのキーで対象を指定し、表示位置はset_position()メソッド、表示の有無はset_visible()で操作する。

軸の指定

spines['bottom']とspines['left']は下と左の軸で、軸の値が表示される。

spines['top']とspines['right']は上と右の軸で、ただ線が引かれるだけ。

各軸に対して、set_positon()、set_visible()の各メソッドを実行して、位置や可視／不可視を設定する。

軸の表示・非表示

set_visible(False)で軸を非表示にする。

以下の例では、上の軸と右の軸を非表示にしている。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 1, 50)

fig , axes = plt.subplots(1, 2)

for ax in axes:
    ax.plot(x, 1 - x)

    ax.set_xlim(0, 1)
    ax.set_ylim(0, 1)

    ax.grid()
    ax.set_aspect('equal')

axes[1].spines['right'].set_visible(False)
axes[1].spines['top'].set_visible(False)

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(0, 1, 50)

fig , axes = plt.subplots(1, 2)

for ax in axes:

ax.plot(x, 1 - x)

ax.set_xlim(0, 1)

ax.set_ylim(0, 1)

ax.grid()

ax.set_aspect('equal')

axes[1].spines['right'].set_visible(False)

axes[1].spines['top'].set_visible(False)

plt.show()

ゼロ位置／中央に軸位置を設定

set_position(‘zero’)でゼロの位置に、set_position(‘center’)で描画位置の中央に軸をセットできる。set_visible()と組み合わせて使うケースが多そう。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 2, 50)

fig , axes = plt.subplots(1, 2)

for ax in axes:
    ax.plot(x, 1 - x)

    ax.set_xlim(-1, 2)
    ax.set_ylim(-1, 2)

    ax.set_xticks(np.arange(-1, 2.5, 0.5))
    ax.set_yticks(np.arange(-1, 2.5, 0.5))

    ax.grid()
    ax.set_aspect('equal')

    ax.spines['right'].set_visible(False)
    ax.spines['top'].set_visible(False)

axes[0].spines['bottom'].set_position('zero')
axes[0].spines['left'].set_position('zero')

axes[1].spines['bottom'].set_position('center')
axes[1].spines['left'].set_position('center')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 2, 50)

fig , axes = plt.subplots(1, 2)

for ax in axes:

ax.plot(x, 1 - x)

ax.set_xlim(-1, 2)

ax.set_ylim(-1, 2)

ax.set_xticks(np.arange(-1, 2.5, 0.5))

ax.set_yticks(np.arange(-1, 2.5, 0.5))

ax.grid()

ax.set_aspect('equal')

ax.spines['right'].set_visible(False)

ax.spines['top'].set_visible(False)

axes[0].spines['bottom'].set_position('zero')

axes[0].spines['left'].set_position('zero')

axes[1].spines['bottom'].set_position('center')

axes[1].spines['left'].set_position('center')

plt.show()

軸の位置の数値指定

set_position()の引数として、タプルで('指定方法', 値)の形で与える。

指定方法	値
data	各軸を配置するx、yの値。
outward	単位はポイントで、正なら描画領域の内側、負なら外側に配置。
axes	描画領域の高さ・幅に対する割合。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 50)

fig , axes = plt.subplots(2, 2)

for row in axes:
    for ax in row:
        ax.plot(x, 10 - x)

        ax.set_xlim(0, 10)
        ax.set_ylim(0, 10)

        ax.set_xticks(np.arange(0, 12, 2))
        ax.set_yticks(np.arange(0, 12, 2))

        ax.grid()
        ax.set_aspect('equal')

ax = axes[0, 1]
ax.set_title('data')
ax.spines['bottom'].set_position(('data', 1))
ax.spines['left'].set_position(('data', 2))
ax.spines['top'].set_position(('data', 7))
ax.spines['right'].set_position(('data', 8))

ax = axes[1, 0]
ax.set_title('outward')
ax.spines['bottom'].set_position(('outward', 5))
ax.spines['left'].set_position(('outward', -10))
ax.spines['top'].set_position(('outward', 15))
ax.spines['right'].set_position(('outward', -20))

ax = axes[1, 1]
ax.set_title('axes')
ax.spines['bottom'].set_position(('axes', 0.1))
ax.spines['left'].set_position(('axes', 0.2))
ax.spines['top'].set_position(('axes', 0.7))
ax.spines['right'].set_position(('axes', 0.8))

fig.subplots_adjust(hspace=0.5)
plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(0, 10, 50)

fig , axes = plt.subplots(2, 2)

for row in axes:

for ax in row:

ax.plot(x, 10 - x)

ax.set_xlim(0, 10)

ax.set_ylim(0, 10)

ax.set_xticks(np.arange(0, 12, 2))

ax.set_yticks(np.arange(0, 12, 2))

ax.grid()

ax.set_aspect('equal')

ax = axes[0, 1]

ax.set_title('data')

ax.spines['bottom'].set_position(('data', 1))

ax.spines['left'].set_position(('data', 2))

ax.spines['top'].set_position(('data', 7))

ax.spines['right'].set_position(('data', 8))

ax = axes[1, 0]

ax.set_title('outward')

ax.spines['bottom'].set_position(('outward', 5))

ax.spines['left'].set_position(('outward', -10))

ax.spines['top'].set_position(('outward', 15))

ax.spines['right'].set_position(('outward', -20))

ax = axes[1, 1]

ax.set_title('axes')

ax.spines['bottom'].set_position(('axes', 0.1))

ax.spines['left'].set_position(('axes', 0.2))

ax.spines['top'].set_position(('axes', 0.7))

ax.spines['right'].set_position(('axes', 0.8))

fig.subplots_adjust(hspace=0.5)

plt.show()

matplotlib.pyplot.contour/contourf – 等高線

2020-02-09 / tau / コメントする

contour～コンターライン

matplotlib.pyplot.contour()は2次元平面上のコンター(等値線)を描く。x, yの値から計算されたzの値が等しい点を曲線で結ぶ。

x, y, zの指定方法

引数がzのみの場合

以下のコードでは、zの値のみを2次元配列で与えている。この場合は、配列zのインデックスの値が座標値となる(行番号0～20がy座標、列番号0～16がx座標)。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 17)
y = np.linspace(-1.2, 1.2, 21)
z = np.array([[u*u + v*v for u in x] for v in y])

fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(111)
ax.contour(z)
ax.set_aspect('equal')
plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 17)

y = np.linspace(-1.2, 1.2, 21)

z = np.array([[u*u + v*v for u in x] for v in y])

fig = plt.figure(figsize=(5, 5))

ax = fig.add_subplot(111)

ax.contour(z)

ax.set_aspect('equal')

plt.show()

引数に1次元のx, yを指定

以下のコードでは、x, yを1次元配列で引数として渡している。この場合は、x, yの値が座標値として用いられる(x座標が-1～1、y座標が-1.2～1.2)。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 17)
y = np.linspace(-1.2, 1.2, 21)
z = np.array([[u*u + v*v for u in x] for v in y])

fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(111)
ax.contour(x, y, z)
ax.set_aspect('equal')
plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 17)

y = np.linspace(-1.2, 1.2, 21)

z = np.array([[u*u + v*v for u in x] for v in y])

fig = plt.figure(figsize=(5, 5))

ax = fig.add_subplot(111)

ax.contour(x, y, z)

ax.set_aspect('equal')

plt.show()

meshgridを使う方法

最も一般的な方法。x, yの1次元配列からnumpy.meshgrid()メソッドでそれぞれの2次元配列を生成し、それらを使ってzを計算するとともに、contour()の引数に与える。結果は上と同じになる。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 17)
y = np.linspace(-1.2, 1.2, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y

fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(111)
ax.contour(x, y, z)
ax.set_aspect('equal')
plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 17)

y = np.linspace(-1.2, 1.2, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y

fig = plt.figure(figsize=(5, 5))

ax = fig.add_subplot(111)

ax.contour(x, y, z)

ax.set_aspect('equal')

plt.show()

ラベル

コンターの値をラベル表示させる。手順としては、contour()で描画した際の戻り値のオブジェクトを保存しておき、それをclabel()の引数として与える。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
z = np.array([[u*u + v*v for u in x] for v in y])

fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(111)
cntr = ax.contour(x, y, z)
ax.clabel(cntr)
ax.set_aspect('equal')
plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

z = np.array([[u*u + v*v for u in x] for v in y])

fig = plt.figure(figsize=(5, 5))

ax = fig.add_subplot(111)

cntr = ax.contour(x, y, z)

ax.clabel(cntr)

ax.set_aspect('equal')

plt.show()

コンターレベル

コンターの数をlevelsで指定する。数値で指定する方法と、配列等で指定する方法があるが、数値で指定する方法は条件によって期待した結果にならないことがある。

数値による指定

1つの整数値で指定する場合、ドキュメンテーションでは以下のように書かれている。

“If an int n, use n data intervals; i.e. draw n+1 contour lines. The level heights are automatically chosen.”

すなわち、n個の間隔に対してn+1本のコンターが描かれることになっている。

以下のコードを実行してみる。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z1 = x + y
z2 = x + y + 2

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
cntr = ax1.contour(x, y, z1, levels=5)
ax1.clabel(cntr)
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
cntr = ax2.contour(x, y, z2, levels=5)
ax2.clabel(cntr)
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z1 = x + y

z2 = x + y + 2

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

cntr = ax1.contour(x, y, z1, levels=5)

ax1.clabel(cntr)

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

cntr = ax2.contour(x, y, z2, levels=5)

ax2.clabel(cntr)

ax2.set_aspect('equal')

plt.show()

範囲と関数の関係によって、同じlevels=5を指定しているのにコンターの本数が異なる。ドキュメント通りなら、5つの間隔に対して6本のコンターが描かれるはずだが、左は6つの間隔に対して5本、右は5つの間隔に対して4本。

右の図の場合は左下でz2 = 0、右上でz2 = 4となり、コーナーの点がコンターに含まれているとすると勘定は合う。左はこれが合わないが、桁落ちなのかゼロが含まれるときに挙動が違うのか、よくわからない。

以下のようにlevels=1とした場合も、挙動が一定しない。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-2, 2, 21)
y = np.linspace(-2, 2, 21)
x, y = np.meshgrid(x, y)
z1 = x * x + y * y - 1
z2 = x * x + y * y

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
cntr = ax1.contour(x, y, z1, levels=1)
ax1.clabel(cntr)
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
cntr = ax2.contour(x, y, z2, levels=1)
ax2.clabel(cntr)
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-2, 2, 21)

y = np.linspace(-2, 2, 21)

x, y = np.meshgrid(x, y)

z1 = x * x + y * y - 1

z2 = x * x + y * y

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

cntr = ax1.contour(x, y, z1, levels=1)

ax1.clabel(cntr)

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

cntr = ax2.contour(x, y, z2, levels=1)

ax2.clabel(cntr)

ax2.set_aspect('equal')

plt.show()

配列等による指定

levelsをリストなどで指定すると、その要素で指定された値のコンターを描く。要素は昇順でなければならない(昇順でない場合は実行時エラー)。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x + y

fig = plt.figure(figsize=(6, 5))

ax = fig.add_subplot(111)
cntr = ax.contour(x, y, z, levels=[-2, -1, 0, 0.5, 1, 1.25, 1.5])
ax.clabel(cntr)
ax.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x + y

fig = plt.figure(figsize=(6, 5))

ax = fig.add_subplot(111)

cntr = ax.contour(x, y, z, levels=[-2, -1, 0, 0.5, 1, 1.25, 1.5])

ax.clabel(cntr)

ax.set_aspect('equal')

plt.show()

線のデザイン

linewidths～線の太さ

線の太さはlinewidthsで指定する。1つの数値で指定した場合は全てのコンターラインに適用、配列等で指定した場合は、サイクリックにその太さが適用される。赤字で示したように、引数名の最後に”s”が着く点に注意。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
ax1.contour(x, y, z, linewidths=3.0)
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
ax2.contour(x, y, z, linewidths=[1, 2, 3])
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

ax1.contour(x, y, z, linewidths=3.0)

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

ax2.contour(x, y, z, linewidths=[1, 2, 3])

ax2.set_aspect('equal')

plt.show()

linestyles～線のスタイル

linestilesで、線のスタイルをスタイル名で指定する。複数指定した場合はサイクリックに適用される。

なお、線の色に単色を用いた場合は、負の値のコンターラインが破線('dashed')で描かれるが、この例は次のcolorsのところで示す。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
ax1.contour(x, y, z, linestyles='dashdot')
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
ax2.contour(x, y, z, linestyles=['solid', 'dashed', 'dashdot', 'dotted'])
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

ax1.contour(x, y, z, linestyles='dashdot')

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

ax2.contour(x, y, z, linestyles=['solid', 'dashed', 'dashdot', 'dotted'])

ax2.set_aspect('equal')

plt.show()

colors～線の色

コンターラインの色はcolorsで指定する。配列等で指定するのが標準だが、単色の場合は配列化せず色名のみで指定可能。ただしその場合は、red, blue等の色名による指定方法のみ。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
ax1.contour(x, y, z, colors='blue')
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
ax2.contour(x, y, z, colors=['blue', 'green', 'red'])
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

ax1.contour(x, y, z, colors='blue')

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

ax2.contour(x, y, z, colors=['blue', 'green', 'red'])

ax2.set_aspect('equal')

plt.show()

単色指定の場合、負の値に対するコンターは破線となる。

If linestyles is None, the default is ‘solid’ unless the lines are monochrome. In that case, negative contours will take their linestyle fromrcParams["contour.negative_linestyle"] = 'dashed' setting.

cmap～カラーマップ

線の色にcolormapを適用できる。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
ax1.contour(x, y, z, cmap='summer')
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
ax2.contour(x, y, z, cmap='seismic')
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

ax1.contour(x, y, z, cmap='summer')

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

ax2.contour(x, y, z, cmap='seismic')

ax2.set_aspect('equal')

plt.show()

alpha～透過度

線によるコンターではあまり意味がないが、線の透過度を0 ～1の実数で指定できる。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z1 = (x + 0.5)**2 + y * y - 1
z2 = (x - 0.5)**2 + y * y - 1

fig = plt.figure(figsize=(8, 5))

ax = fig.add_subplot(111)
ax.contour(x, y, z1, alpha=0.5)
ax.contour(x, y, z2, alpha=0.5)
ax.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z1 = (x + 0.5)**2 + y * y - 1

z2 = (x - 0.5)**2 + y * y - 1

fig = plt.figure(figsize=(8, 5))

ax = fig.add_subplot(111)

ax.contour(x, y, z1, alpha=0.5)

ax.contour(x, y, z2, alpha=0.5)

ax.set_aspect('equal')

plt.show()

contourf～色付きのコンターエリア

contourfはコンターラインで区切られた各エリアを色付けする。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(111)
ax.contourf(x, y, z)
ax.set_aspect('equal')
plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(5, 5))

ax = fig.add_subplot(111)

ax.contourf(x, y, z)

ax.set_aspect('equal')

plt.show()

ラベル

contourfでラベルを付けることはあまり想定されないが、contourと同じようにすると図が崩れてしまう。contourとcontourfの合わせ技がよい。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
cntr = ax1.contourf(x, y, z)
ax1.clabel(cntr, colors='black')
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
ax2.contourf(x, y, z)
cntr = ax2.contour(x, y, z)
ax2.clabel(cntr, colors='black')
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

cntr = ax1.contourf(x, y, z)

ax1.clabel(cntr, colors='black')

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

ax2.contourf(x, y, z)

cntr = ax2.contour(x, y, z)

ax2.clabel(cntr, colors='black')

ax2.set_aspect('equal')

plt.show()

デザイン

cmap～カラーマップ

コンターエリアの色分けにカラーマップを指定できる。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)
ax1.contourf(x, y, z, cmap='seismic')
ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)
ax2.contourf(x, y, z, cmap='cividis')
ax2.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z = x * x + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(121)

ax1.contourf(x, y, z, cmap='seismic')

ax1.set_aspect('equal')

ax2 = fig.add_subplot(122)

ax2.contourf(x, y, z, cmap='cividis')

ax2.set_aspect('equal')

plt.show()

alpha～透過度

透過度を1未満に設定して、透過させることができる。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)
y = np.linspace(-1, 1, 21)
x, y = np.meshgrid(x, y)
z1 = (x + 0.5)**2 + y * y - 1
z2 = (x - 0.5)**2 + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(131)
ax1.contourf(x, y, z1, cmap='autumn', alpha=0.5)
ax1.set_aspect('equal')

ax2 = fig.add_subplot(132)
ax2.contourf(x, y, z2, cmap='winter', alpha=0.5)
ax2.set_aspect('equal')

ax3 = fig.add_subplot(133)
ax3.contourf(x, y, z1, cmap='autumn', alpha=0.5)
ax3.contourf(x, y, z2, cmap='winter', alpha=0.5)
ax3.set_aspect('equal')

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-1, 1, 21)

y = np.linspace(-1, 1, 21)

x, y = np.meshgrid(x, y)

z1 = (x + 0.5)**2 + y * y - 1

z2 = (x - 0.5)**2 + y * y - 1

fig = plt.figure(figsize=(12, 5))

ax1 = fig.add_subplot(131)

ax1.contourf(x, y, z1, cmap='autumn', alpha=0.5)

ax1.set_aspect('equal')

ax2 = fig.add_subplot(132)

ax2.contourf(x, y, z2, cmap='winter', alpha=0.5)

ax2.set_aspect('equal')

ax3 = fig.add_subplot(133)

ax3.contourf(x, y, z1, cmap='autumn', alpha=0.5)

ax3.contourf(x, y, z2, cmap='winter', alpha=0.5)

ax3.set_aspect('equal')

plt.show()

matplotlib.pyplot – 主なグラフの設定概要

2020-01-05 / tau / コメントする

概要

matlib.pyplotの各グラフの一般的な設定をまとめておく。なお、カラー、線のスタイル、マーカーの設定はそれぞれ以下を参照

各グラフの主な設定

scatter～散布図

scatter(x, y): デフォルト設定で散布図を描く。
scatter(x, y, marker=markerstyle, c=color): マーカーの種類と色を指定。色指定はc=/color=のどちらでもよい。
scatter(x, y, marker=markerstyle, s=size, color=color): マーカーのサイズも指定

plot～折れ線

plot(x, y): デフォルト設定で折れ線グラフを描く。
plot(x, y, c=color): 線の色を設定。色指定はc=/color=のどちらでもよい
plot(x, y, linestyle=linestyle, linewidth=width): 線のスタイルと太さを設定。
plot(x, y, marker=markerstyle, markersize=size): マーカーのスタイルとサイズを設定。
plot(x, y, styles[, x, y, styles]...): スタイル文字列を使う。スタイル文字列は、たとえば’g^--‘のように表現。複数の線をまとめて指定可能。

barh～横棒グラフ

barh(y, width): ラベルと値を指定して棒グラフを描く。
hist(y, width, fc, ec, linewidth): 塗りつぶし色、枠の色や太さを指定。

hist～ヒストグラム

hist(data, bins=bins, label=label): ビン数を指定してヒストグラムを描く。
hist(data, bins=bins, range=(min, max), color=color, edgecolor=color, linewidgh=width): ビン数、レンジ、塗りつぶし色、枠線の色を指定。
hist(data, density=True): 頻度表示で設定。デフォルトは度数表示
hist(data, alpha=alpha): 複数のヒストグラムを重ねて描く場合の透明度をそれぞれ指定。各グラフのビン幅を揃えたいときは同じレンジを指定。
hist([data1, data2,...], bins=bins, color=[c1, c2,...], label=[l1, l2,...]): 複数のヒストグラムを重ねずに並べて描く。
hist([data1, data2,...], bins=bins, stacked=True): 複数のヒストグラムを積み上げて描く。

円グラフ

pie(data, radius=r, counterclock=False, startangle=90, labels=labels, autopct=pctstring): ラベルとパーセンテージの書式を指定して円グラフを描く。radiusは通常デフォルトの1を使う。counterclockとstartangleはいちいち指定しないといけない。autopctの書式の基本形は、たとえば"%.1f"。ラベルとパーセンテージの位置を、半径の割合で指定可能(labeldistance, pctdistance)
pie(data, counteerclock=False, startangle=90, explode=explodelist): explodeで各データの中心からの位置をリスト指定。同じ小さい値を全データにい指定するとデータ間に隙間ができる。特定データの値を大きくすると、そのデータのみ飛び出る。
pie(data, counteerclock=False, startangle=90, wedgegroups={'linewidgh':width, 'edgecolor'=color): wedgegroupsに辞書を指定して、枠線の幅と色を指定。

ドーナツ型の

matplotlib – linestyle～線のスタイル

2020-01-03 / tau / コメントする

文字列／記号による指定

一般に使われる4つのスタイルについては、文字列あるいは記号で指定。

`"solid"`	“-“	連続線
`"dashed"`	“–“	破線
`"dotted"`	“:”	点線
`"dashdot"`	“-.”	一点鎖線

文字列指定の例

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)

x = np.array([0, 1])
y = np.array([0, 1])

ax.plot(x, y, linestyle="solid")
ax.plot(x, y+1, linestyle="dashed")
ax.plot(x, y+2, linestyle="dotted")
ax.plot(x, y+3, linestyle="dashdot")

plt.show()

import numpy as np

import matplotlib.pyplot as plt

fig = plt.figure()

ax = fig.add_subplot(111)

x = np.array([0, 1])

y = np.array([0, 1])

ax.plot(x, y, linestyle="solid")

ax.plot(x, y+1, linestyle="dashed")

ax.plot(x, y+2, linestyle="dotted")

ax.plot(x, y+3, linestyle="dashdot")

plt.show()

記号指定の例

ax.plot(x, y, linestyle="-")
ax.plot(x, y+1, linestyle="--")
ax.plot(x, y+2, linestyle=":")
ax.plot(x, y+3, linestyle="-.")

ax.plot(x, y, linestyle="-")

ax.plot(x, y+1, linestyle="--")

ax.plot(x, y+2, linestyle=":")

ax.plot(x, y+3, linestyle="-.")

実行結果

タプルによる任意のパターン指定

タプルによって、任意のパターンを設定することができる。

(offset, (fg, bg, fg, bg, ...))

第2要素のタプルの中で、前景色の長さと背景色の長さを交互に指定。第1要素は病か開始のオフセット。

ax.plot(x, y, linestyle=(0, (1, 0)))
ax.plot(x, y+1, linestyle=(0, (0, 1)))
ax.plot(x, y+2, linestyle=(0, (1, 1)))
ax.plot(x, y+3, linestyle=(0, (5, 1)))
ax.plot(x, y+4, linestyle=(0, (5, 1, 1, 1)))
ax.plot(x, y+5, linestyle=(0, (5, 3, 1, 3, 1, 3)))
ax.plot(x, y+6, linestyle=(10, (5, 3, 1, 3, 1, 3)))

ax.plot(x, y, linestyle=(0, (1, 0)))

ax.plot(x, y+1, linestyle=(0, (0, 1)))

ax.plot(x, y+2, linestyle=(0, (1, 1)))

ax.plot(x, y+3, linestyle=(0, (5, 1)))

ax.plot(x, y+4, linestyle=(0, (5, 1, 1, 1)))

ax.plot(x, y+5, linestyle=(0, (5, 3, 1, 3, 1, 3)))

ax.plot(x, y+6, linestyle=(10, (5, 3, 1, 3, 1, 3)))

matplotlib – color～色指定

2020-01-03 / tau / コメントする

概要

とりあえずのメモ書き。

1文字での指定

`'b'`	blue/青	■
`'g'`	green/緑	■
`'r'`	red/赤	■
`'c'`	cyan/シアン	■
`'m'`	magenta/マゼンタ	■
`'y'`	yellow/黄	■
`'k'`	black/黒	■
`'w'`	white/白	□

カラーネーム

matplotlibのカラーネーム

matplotlibのドキュメンテーション

グラフの標準色

各種グラフのデフォルトで用いられる標準色がある。

グレー表示

濃淡のレベルを”文字列で”指定。

color=['0.0', '0.2', '0.4', '0.6', '0.8', '1.0']

matplotlib – markers～マーカースタイル

2020-01-03 / tau / コメントする

グラフ表示などで使えるマーカーの種類は以下の通り。

スタイル指定は以下の通り。

“.”	point	・
“,”	pixel	■
“o”	circle	●
“v”	triangle_down	▼
“^”	triangle_up	▲
“<“	triangle_left	◀
“>”	triangle_right	▶
“1”	tri_down	下向き三又
“2”	tri_up	上向き三又
“3”	tri_left	左向き三又
“4”	tri_right	右向き三又
“8”	octagon	⯃
“s”	square	■
“p”	pentagon	⬟
“*”	star	★
“h”	hexagon1	⬢
“H”	hexagon2	⬣
“+”	plus	＋
“x”	x	×
“D”	diamond	◆
“d”	thin_diamond	♦
“\|”	vline	｜
“_”	hline	―
“P”	plus_filled	✚
“X”	x_filled	✖
0	tickleft	╴
1	tickright	╶
2	tickup	╵
3	tickdown	╷
4	caretleft	⯇
5	caretright	⯈
6	caretup	⯅
7	caretdown	⯆
8	caretleftbase	中心ベース左キャレット
9	caretrightbase	中心ベース右キャレット
10	caretupbase	中心ベース上キャレット
11	caretdownbase	中心ベース下キャレット
“None”	マーカーなし
None	マーカーなし	●が表示
” “	マーカーなし
“”	マーカーなし

matplotlib.pyplot.axes – 縦横比をそろえる

2020-01-02 / tau / コメントする

pyplot.axesでグラフを表示する際に、縦横比を1:1に揃えたい場合。

axes.set_aspect('equal')

import numpy as np
import matplotlib.pyplot as plt

num_points = 50
x = np.zeros(num_points)
y = np.zeros(num_points)

for n in range(0, 50):
    x[n] = np.random.rand()
    y[n] = np.random.rand()

fig = plt.figure()
ax1 = fig.add_subplot(211)
ax2 = fig.add_subplot(212)

ax1.grid(True)
ax2.grid(True)

ax2.set_aspect('equal')

ax1.scatter(x, y)
ax2.scatter(x, y)

plt.show()

import numpy as np

import matplotlib.pyplot as plt

num_points = 50

x = np.zeros(num_points)

y = np.zeros(num_points)

for n in range(0, 50):

x[n] = np.random.rand()

y[n] = np.random.rand()

fig = plt.figure()

ax1 = fig.add_subplot(211)

ax2 = fig.add_subplot(212)

ax1.grid(True)

ax2.grid(True)

ax2.set_aspect('equal')

ax1.scatter(x, y)

ax2.scatter(x, y)

plt.show()