umpy库
umpyumpy库中的where函数。该函数的作用是返回满足条件的元素的索引。
umpy库获取矩阵中的1
portumpyp
创建一个3×3的矩阵atrixp.array([[ 1], [ 0], [ 0]])
使用where函数获取矩阵中的1dicespatrix == 1)
tdices)
运行上述代码，输出结果为
(array([ 2]), array([ 2, 0]))
可以看到，where函数返回了两个数组，分别表示满足条件的元素的行索引和列索引。例如，个元素( 0)表示矩阵中行列的元素是1。
方法2使用列表推导式
umpy还提供了许多其他的方法来获取矩阵中的1。其中，列表推导式是一种简单而有效的方法。列表推导式可以用于从一个列表中获取满足特定条件的元素，而矩阵本质上就是一个嵌套的列表。
下面是一个使用列表推导式获取矩阵中的1的例子
创建一个3×3的矩阵atrix = [[ 1], [ 0], [ 0]]
使用列表推导式获取矩阵中的1dicesgeatrixgeatrixatrix[i][j] == 1]
tdices)
运行上述代码，输出结果为
[( 0), ( 2), ( 1), (2, 0)]
可以看到，使用列表推导式获取矩阵中的1也很简单。该方法通过两个嵌套的for循环遍历矩阵中的所有元素，并使用if语句检查每个元素是否等于1。如果是，则将元素的行索引和列索引添加到一个列表中。
umpy库中的where函数可以快速地获取满足条件的元素的索引。而使用列表推导式则可以在不使用任何库的情况下获取矩阵中的1。无论您选择哪种方法，都可以轻松地获取矩阵中的1，以便进行进一步的数据处理和分析。