MatplotlibでマスクされたNaN値をプロットする方法は？

MatplotlibでマスクされたNaN値をプロットするには、次の手順を実行できます-

• 図のサイズを設定し、サブプロット間およびサブプロットの周囲のパディングを調整します。
• xを作成します およびy numpyを使用したデータポイント。
• x2を取得する およびy2 y> 0.7のようなデータポイント 。
• マスクされるy3 y> 0.7のようなデータポイント 。
• マスクy3 NaN値を使用します。
• xをプロットします 、 y 、 y2 、 y3 およびy4 plot（）を使用する メソッド。
• プロットに凡例を配置します。
• プロットのタイトルを設定します。
• 図を表示するには、 show（）を使用します メソッド。

例

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(-np.pi/2, np.pi/2, 31)
y = np.cos(x)**3

# 1) remove points where y > 0.7
x2 = x[y <= 0.7]
y2 = y[y <= 0.7]

# 2) mask points where y > 0.7
y3 = np.ma.masked_where(y > 0.7, y)

# 3) set to NaN where y > 0.7
y4 = y.copy()
y4[y3 > 0.7] = np.nan

plt.plot(x*0.1, y, 'o-', color='lightgrey', label='No mask')
plt.plot(x2*0.4, y2, 'o-', label='Points removed')
plt.plot(x*0.7, y3, 'o-', label='Masked values')
plt.plot(x*1.0, y4, 'o-', label='NaN values')
plt.legend()

plt.title('Masked and NaN data')

plt.show()

出力