np.argwhere与np.where区别

np.argwhere和np.where都是NumPy中用于查找数组中满足某个条件的元素索引的函数。它们的主要区别在于返回结果的形式和用法上。

np.where(condition)函数返回满足条件的元素的索引，以元组的形式表示。每个元组表示一个满足条件的元素的索引。如果数组是一维的，那么返回的结果是一个一维数组；如果数组是多维的，那么返回的结果是一个元组，其中每个元素是一个一维数组。

np.argwhere(condition)函数返回满足条件的元素的索引，以二维数组的形式表示。每一行表示一个满足条件的元素的索引。其中，每一行的第一个元素表示该元素在第0维的索引，第二个元素表示该元素在第1维的索引，以此类推。返回的结果是一个二维数组。

简而言之，np.where返回满足条件的元素索引的元组或数组，而np.argwhere返回满足条件的元素索引的二维数组。