Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีการพล็อตค่ามาสก์และ NaN ใน Matplotlib?


ในการพล็อตค่ามาสก์และค่า NaN ใน Matplotlib เราสามารถทำตามขั้นตอนต่อไปนี้ -

  • กำหนดขนาดรูปและปรับช่องว่างภายในระหว่างและรอบๆ แผนผังย่อย
  • สร้าง x และ จุดข้อมูลโดยใช้ตัวเลข
  • รับ x2 และ ปี2 จุดข้อมูลดังกล่าวว่า y> 0.7 .
  • สวมหน้ากาก y3 จุดข้อมูลดังกล่าวว่า y> 0.7 .
  • หน้ากาก y3 ด้วยค่า NaN
  • พล็อต x , , ปี2 , 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()

ผลลัพธ์

วิธีการพล็อตค่ามาสก์และ NaN ใน Matplotlib?