查询矩阵中的一行

Question

我有这个代码：

X = tfidf.fit_transform(data['Content']) # the matrix articles x max_features(=words)
dense = X.toarray()
print dense[432]

输出为：

[ 0.          0.          0.45446165  0.          0.22281507  0.
  0.27551889  0.37350294  0.          0.72691331]

能否反方向求矩阵，即这一行的index是多少？

我显然可以创建一个 for 循环并手动搜索该行。

Answer 1

如果您能够使用密集矩阵，那么它就是一个简单的数组比较，例如

np.where((dense == row).all(axis=1))[0]

Query for a row in a matrix