列表理解中的多条件语句

Question

这是我的代码，我想知道是否可以使用列表理解来执行相同的操作（计算行中的集群，并输出长度为df.shape[0]的列表）。同一个集群编号至少有两行，但可以多行，并且可以循环。我试过了，但没能弄明白。 有什么建议吗

我的代码：

    import pandas as pd

    cluster_global = 0
    cluster_relativo = 0
    cluster_index = []
    for index, row in df.iterrows():
        if row['cluster'] == cluster_relativo:
            cluster_index.append(cluster_global)
        elif row['cluster'] == (cluster_relativo + 1):
            cluster_global += 1
            cluster_relativo += 1
            cluster_index.append(cluster_global)
        elif row['cluster'] == 0:
            cluster_global += 1
            cluster_relativo = 0
            cluster_index.append(cluster_global)

数据帧看起来像

^{tb1}$

Answer 1

你想要这个吗

from itertools import groupby

result = [0 if index == 0 and key == 0
          else index
          for index, (key, group) in enumerate(groupby(my_values))
          for _ in group
          ]

print(result)

通过-df['cluster']替换列表中的my_值。值。检验