Vẽ giá trị bị che (masked) và NaN trong Matplotlib

Các bước

1. Thiết lập kích thước hình và padding.
2. Tạo dữ liệu x, y bằng numpy.
3. Lấy x2, y2 nơi y <= 0.7 (loại bỏ điểm).
4. Tạo y3 bị che (masked) nơi y > 0.7.
5. Tạo y4 gán NaN nơi y > 0.7.
6. Vẽ 4 trường hợp: không che, loại bỏ điểm, masked, NaN.
7. Thêm legend và tiêu đề.
8. Gọi show().

Ví dụ

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) Loại bỏ điểm y > 0.7
x2 = x[y <= 0.7]
y2 = y[y <= 0.7]

# 2) Che điểm y > 0.7
y3 = np.ma.masked_where(y > 0.7, y)

# 3) Gán NaN nơi 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()

Kết quả