python - 如何根据pandas中其他列表的值获取不连续列表？

Question

我正在尝试向数据框提供数字列表。我想要的结果如下:

Unit        Ida            1      Parcel 1    2Parcel 2    2Parcel 3    2            3            4Parcel 1    5Parcel 2    5

我使用的第一个代码如下:

    Address['Ida'] = ''    Ida = 1    Address['Ida'][0] = Ida    for x in range(len(Address)-1):        if str(Address['Unit'][x+1]) == ('Parcel 1' or ''):            Ida = Ida + 1            Address['Ida'][x+1] = Ida        else:            Address['Ida'][x+1] = Ida

但是我的代码速度非常非常慢。那么在pandas中，有没有更高效的方法呢？

Answer 1

IIUC 使用 cumsum 和 shift

((df.Unit=='')|(df.Unit=='Parcel1')).cumsum()
Out[129]: 
0    1
1    2
2    2
3    2
4    3
5    4
6    5
7    5
Name: Unit, dtype: int32