使用reduce、map或其他函数来避免python中的for循环

Question

我有一个程序用于计算距离，然后应用 k-means 算法。我在一个小列表上进行了测试，它运行良好且快速，但是，我的原始列表非常大（> 5000），所以它需要很长时间，我最终终止了运行。我可以使用 outer() 或任何其他并行函数并将其应用于距离函数以使其更快吗？ 在我拥有的小集合上：

strings = ['cosine cos', 'cosine', 'cosine???????', 'l1', 'l2', 'manhattan']

其距离3D数组返回如下：

[[[ 0.          0.25        0.47826087  1.          1.          0.89473684]
  [ 0.25        0.          0.36842105  1.          1.          0.86666667]
  [ 0.47826087  0.36842105  0.          1.          1.          0.90909091]
  [ 1.          1.          1.          0.          0.5         1.        ]
  [ 1.          1.          1.          0.5         0.          1.        ]
  [ 0.89473684  0.86666667  0.90909091  1.          1.          0.        ]]]

上面数组的每一行代表字符串列表中一项的距离。我使用 for 循环的方法是：

strings = ['cosine cos', 'cosine', 'cosine???????', 'l1', 'l2', 'manhattan']


data1 = []


for j in range(len(np.array(list(strings)))):

     for i in range(len(strings)):
       data1.append(1-Levenshtein.ratio(np.array(list(strings))[j], np.array(list(strings))[i]))

#n =(map(Levenshtein.ratio, strings))
#n =(reduce(Levenshtein.ratio, strings))
#print(n)



k=len(strings)
data2=np.asarray(data1)
arr_3d = data2.reshape((1,k,k))
print(arr_3d)

arr_3d 是上面的数组。如何使用 outer() 或 map() 中的任何一个来替换上面的 for 循环，因为当列表 strings 很大时，它需要几个小时，甚至从未得到结果。我很感激帮助。 Levenshtein.ratio 是python的内置函数。

Answer 1

import numpy as np 

strings = ['cosine cos', 'cosine', 'cosine???????', 'l1', 'l2', 'manhattan']

k=len(strings)

data = np.zeros((k,k))

for i,string1 in enumerate(strings):
    for j,string2 in enumerate(strings):
        data[i][j] = 1-Levenshtein.ratio(string1, string2)

print data

这里map 或reduce 没有任何好处，循环需要像@user2357112 提到的那样运行，但是，这更干净并且应该运行得更快，因为它避免了您一直使用的np.array(list(strings))。