NaN散点图中作为特殊颜色的pyplot值

import numpy as np
import matplotlib.pyplot as plt

# 示例数据
x = np.random.rand(10)
y = np.random.rand(10)
# 故意引入一些NaN值
x[np.random.randint(0, 10, 2)] = np.nan
y[np.random.randint(0, 10, 2)] = np.nan

# 创建散点图
plt.scatter(x, y, c='blue', label='Valid Data')

# 找出NaN值的位置
nan_mask = np.isnan(x) | np.isnan(y)
nan_x = x[nan_mask]
nan_y = y[nan_mask]

# 使用特殊颜色标记NaN值
plt.scatter(nan_x, nan_y, c='red', label='NaN Data')

# 添加图例和标签
plt.legend()
plt.xlabel('X Axis')
plt.ylabel('Y Axis')
plt.title('Scatter Plot with NaN Values Highlighted')

# 显示图表
plt.show()