如何处理python中的迭代器？

Question

大家好，我正在尝试使用 nx.jaccard_coefficient 获得的一个未来创建一个列表 (pd.Series)，问题是我得到了一个迭代器，我不知道如何提取我应该获得的结果。 该函数应该返回一个带有 3 个参数 u,v,p 的迭代器。

所以我创建了一个理解列表来提取一些边集的数据。

 future_connections["Jaccard"] = [nx.jaccard_coefficient(G,edge) for edge in 
                                  future_connections['index']]

我得到以下结果

            Future Connection       index    a    b  Common_Neighbors  \
(6, 840)                  0.0    (6, 840)    6  840                 9   
(4, 197)                  0.0    (4, 197)    4  197                 2   
(620, 979)                0.0  (620, 979)  620  979                 0   
(519, 872)                0.0  (519, 872)  519  872                 2 
               Jaccard  
(6, 840)    <generator object jaccard_coefficient.<locals>...  
(4, 197)    <generator object jaccard_coefficient.<locals>...  
(620, 979)  <generator object jaccard_coefficient.<locals>...  
(519, 872)  <generator object jaccard_coefficient.<locals>...  

关于如何提取生成器的值的任何想法？？？

谢谢。

Answer 1

如果你的函数nx.jaccard_coefficient返回一个包含 3 个元素的元组/列表，试试这个。

future_connections["Jaccard"] = [
  tuple(nx.jaccard_coefficient(G,edge)) for edge in  future_connections['index']]

Answer 2

请尝试以下操作：

future_connections["Jaccard"] = [p for u,v,p in nx.jaccard_coefficient(G, future_connections.index)]