python列表嵌套循环,如何在Python中的Joblib库中使用嵌套循环

Actual code looks like:

def compute_score(row_list,column_list):

for i in range(len(row_list)):

for j in range(len(column_list)):

tf_score = self.compute_tf(column_list[j],row_list[i])

I am tying to achieve multi-processing i.e. at every iteration of j I want to pool column_list. Since compute_tf function is slow I want to multi-process it.

I've found have to do it using joblib in Python, But I am unable to workaround with nested loops.

Parallel(n_jobs=2)(delayed(self.compute_tf))

This is what is to be achieved.

It would be a great help if any solution on this is provided or any-other solution.

解决方案

Another solution without having to implement a generator function, is to use the nested list comprehension for the generator:

Parallel(n_jobs=2)(delayed(self.compute_tf)(i, j) for j in column_list for i in row_list)

The order will be given as:

[(i, j) for j in range(10) for i in range(10)]