浅谈numpy.where() 的用法和np.argsort()的用法说明

以下是浅谈numpy.where()的用法和np.argsort()的用法说明的攻略：

numpy.where()的用法

在numpy中，可以使用numpy.where()函数来根据条件返回数组中的元素。以下是一些示例：

返回满足条件的元素

可以使用numpy.where()函数来返回满足条件的元素。以下是一个示例：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.where(a > 2)
print(b)

输出：

(array([2, 3, 4]),)

返回满足条件的元素和不满足条件的元素

可以使用numpy.where()函数来返回满足条件的元素和不满足条件的元素。以下是一个示例：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.where(a > 2, a, 0)
print(b)

输出：

[0 0 3 4 5]

np.argsort()的用法说明

在numpy中，可以使用np.argsort()函数来返回数组排序后的索引。以下是一些示例：

返回一维数组排序后的索引

可以使用np.argsort()函数来返回一维数组排序后的索引。以下是一个示例：

import numpy as np

a = np.array([3, 1, 4, 2, 5])
b = np.argsort(a)
print(b)

输出：

[1 3 0 2 4]

返回二维数组按行排序后的索引

可以使用np.argsort()函数来返回二维数组按行排序后的索引。以下是一个示例：

import numpy as np

a = np.array([[3, 1, 4], [2, 5, 0]])
b = np.argsort(a, axis=1)
print(b)

输出：

[[1 0 2]
 [0 2 1]]

总结

这就是浅谈numpy.where()的用法和np.argsort()的用法说明的攻略。numpy.where()函数可以用来根据条件返回数组中的元素，np.argsort()函数可以用来返回数组排序后的索引。希望这篇文章能够帮助您更好地理解numpy中where()和argsort()函数的使用方法。