Python NumPy 获取二维数组特定列的类别值

可以使用 NumPy 中的 unique 函数来获取二维数组中某列的值的类别。

例如，假设有一个二维数组 arr，其中第一列为类别列，可以使用以下代码来获取第一列的类别：

import numpy as np

arr = np.array([[1, 2], [1, 3], [2, 4], [3, 5], [2, 6]])

categories = np.unique(arr[:, 0])

print(categories)

输出结果为：

[1 2 3]

其中，arr[:, 0] 表示获取 arr 数组中所有行的第一列，即类别列。np.unique 函数则返回该列中的不同值。