python - 如何在Python中运行循环并每次返回到不同的列表

Question

我有数据如下

     Gene  Distance
0    A       18
1    B       16
2    C       58
3    D       45
4    E       34

如果两个基因的距离小于 50，则应按如下方式将它们组合(在列表中)

1 A,B
2 C,D,E

循环应该在 B 和 C 之间进行，因为它们之间的距离超过 50。如何在循环和 for 列表中多次创建此类中断。

Answer 1

您可以使用groupby(应该比循环更快..)

df.Gene.groupby(df.Distance.gt(50).cumsum()).apply(list).str.join(',')
Out[347]: 
Distance
0      A,B
1    C,D,E
Name: Gene, dtype: object