请给我一段获得只含0，1的矩阵获得1元素的python代码，要求使用np的isnan函数

你可以使用NumPy库的`isnat`函数结合布尔索引来获取一个只包含0和1的矩阵，并提取出所有1的元素。以下是一个简单的Python代码示例：

import numpy as np

def get_ones(matrix):
    # 判断矩阵中的每个元素是否为NaN（在NumPy中nan被视为缺失值）
    is_one = ~np.isnan(matrix)  # 使用位操作符~转换全0数组为全1数组，非nan变为True
    # 索引得到只包含1的位置
    ones_positions = np.argwhere(is_one)
    # 返回这些位置对应的1元素
    return matrix[ones_positions]

# 示例
matrix_example = np.array([[1, 0, np.nan], [np.nan, 1, 1]])
result = get_ones(matrix_example)
print("原矩阵：", matrix_example)
print("只含1的元素：", result)