Python | numpy matrix.nonzero()

在Python的NumPy包中，matrix.nonzero()函数返回由非零元素的行和列组成的元组。 该函数可以接受一个可选参数，该参数用于定义要查找的元素的数据类型。

语法

numpy.matrix.nonzero()

参数

无

返回值

该函数返回元组，其包含两个数组，分别表示行索引和列索引。

示例

import numpy as np

a = np.matrix([[0, 1, 0], [1, 0, 0], [0, 0, 1]])
print("Non-zero indexes are:")
print(a.nonzero())

输出：

Non-zero indexes are:
(array([0, 1, 2], dtype=int64), array([1, 0, 2], dtype=int64))

这表示在给定矩阵中，第一行的第二个元素、第二行的第一个元素和第三行的第三个元素是非零元素。

b = np.matrix([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("Non-zero indexes are:")
print(b.nonzero())

输出：

Non-zero indexes are:
(array([0, 0, 1, 1, 2, 2], dtype=int64), array([0, 1, 0, 1, 0, 1], dtype=int64))

这表示在给定矩阵中，第一行的第一个、第二个和第三个元素、第二行的第一个、第二个和第三个元素以及第三行的第一个和第二个元素是非零元素。

参考链接

• Python | numpy.matrix.nonzero()