numpy排序与集合运算用法示例

下面我将为大家详细讲解“numpy排序与集合运算用法示例”的完整攻略。

一、numpy排序用法示例

1.1 排序数组

在numpy中，可以使用sort()方法对数组进行排序，其默认排序方式为升序排序。下面是一个示例：

import numpy as np

arr = np.array([1, 5, 3, 9, 7])
arr.sort()
print(arr)

此时会输出：

[1 3 5 7 9]

1.2 排序矩阵

对于矩阵的排序，可以指定按照某一维度进行排序。下面是一个示例：

import numpy as np

arr = np.array([[1, 4], [3, 1], [6, 9], [2, 8]])
arr_sorted = np.sort(arr, axis=0)
print(arr_sorted)

此时会输出：

[[1 1]
 [2 4]
 [3 8]
 [6 9]]

二、numpy集合运算用法示例

2.1 集合求交

可以使用intersect1d()方法求两个数组的交集。下面是一个示例：

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([3, 4, 5, 6, 7])
arr_intersect = np.intersect1d(arr1, arr2)
print(arr_intersect)

此时会输出：

[3 4 5]

2.2 集合求并

可以使用union1d()方法求两个数组的并集。下面是一个示例：

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([3, 4, 5, 6, 7])
arr_union = np.union1d(arr1, arr2)
print(arr_union)

此时会输出：

[1 2 3 4 5 6 7]

以上就是numpy排序与集合运算用法示例的完整攻略，希望对大家有所帮助。