我正在使用我的numpy数组v
进行如下操作,以删除小于等于1的元素,然后选择numpy数组中前3个元素的索引。
for ele in v.toarray()[0].tolist():
if ele <= 1:
useless_index = v.toarray()[0].tolist().index(ele)
temp_list.append(useless_index)
#take top 3 words from each document
indexes =v.toarray()[0].argsort()[-3:]
useful_list = list(set(indexes) - set(temp_list))
然而,我目前使用的代码非常缓慢(因为我有数百万个numpy数组),需要花费数天时间才能运行。是否有更有效的方法在Python中完成相同的操作?
v[v > 1]
怎么样? - Mateen Ulhaq