np.argwhere什么意思
时间: 2023-11-17 21:06:50 浏览: 201
Python Numpy:找到list中的np.nan值方法
5星 · 资源好评率100%
`np.argwhere()` 是 NumPy 库中的一个函数,用于返回数组中非零元素的索引。它返回一个数组,其中每个行为非零元素的索引。例如,对于数组 `arr`,`np.argwhere(arr)` 返回的是一个形状为 `(n, arr.ndim)` 的数组,其中 `n` 是数组中非零元素的数量,`arr.ndim` 是数组的维度数。每一行表示一个非零元素的索引,其中每个元素对应该维度的索引值。
阅读全文