时间戳

首页 > 解决方案 > 检查行是否包含在 CSR 矩阵中

python - 检查行是否包含在 CSR 矩阵中

问题描述

我有一组 CSR 矩阵格式的数据点，以及数据集中每个 CSR 矩阵格式的聚类点列表。我必须检查我的一组数据点并找出它在哪个集群中。我总共有大约 8000 个数据点。

我尝试遍历数据点和集群并使用in关键字：

       for c in cluster:
           if datapoint.toarray() in c.toarray():
               # do stuff

但in无论如何，测试都会返回 true。任何人都有比逐个元素检查更有效的方法吗？

注意：数据集和集群是 CSR 矩阵。toarray()是来自 scipy 库的 csr 矩阵方法。

标签： pythonsparse-matrix

解决方案

推荐阅读