使用 Concat 将系列移位连接到数据框

Question

有一个 DataFrame z 有

['title','module_number','beginners','finishers'] 

列。我需要转移z.beginners 系列，因为最后一个module_number 应该有NaN，并将其作为新列z.shift 加入DataFrame。

我尝试使用pd.concat 来实现。我的代码：

def exp2(df):
    
    fx = df.join(df.beginners[1:].rename('shifted').reset_index()).drop('index', axis =1)
    return fx

def exp3(z,lst):
    
    common = list()
    
    for title in lst:

        df = z[z['title'] == title]
        example = exp2(df)
        common.append(example)
        
    return pd.concat(common)

exp3(z, z.title.unique())

这仅适用于z 中的第一个标题，而不适用于第二个标题。但是，它不适用于任何其他标题：

        title module_number beginners finishers shift
    0   Uu    1             518       458       434
    1   Uu    2             434       406       NaN
    3   Kk    1             627       563       27
    4   Kk    2             521       427       NaN
    5   Rr    1             158       142       NaN
    6   Rr    2             136       106       NaN

有谁知道替代解决方案或我可以如何增强我的代码？

Answer 1

我刚刚做到了没有循环的方式：

shifted = z.beginners.where(z.module_number != 1)[1:]
z.join(shifted.rename('shifted').reset_index()).drop('index',axis = 1)