使用 np.unique Python 后获取元素数

Question

我得到了一组元素。任务是获取唯一元素的数量。我写了以下内容：

 import numpy as np
 tokens1 = set(["a", "b", "c", "c"])
 print(np.unique(tokens1))
 print(np.unique(tokens1).size)

结果是

[{'c', 'b', 'a'}]
1

我如何获得正确的数字 - 3？我应该首先使用 np.unique 以外的东西吗？可能有更好的方法来获得我想要的帽子。

Answer 1

为了获得集合中唯一元素的数量，您可以简单地调用len()该集合。要修改现有代码：

tokens1 = set(["a", "b", "c", "c"])
print(len(tokens1))
# prints: 3

这是因为集合已经删除了重复项。您不需要同时使用 set AND np.unique()。

如果您想np.unique()改用，您可以将代码修改为：

tokens1 = np.unique(["a", "b", "c", "c"])
print(len(tokens1))
# prints: 3