konsekutiv nächster Wert in neue Spalt Pandas Datenrahmen

Question

Ich habe einen Datenrahmen

id value 
v1  100 
v1  200 
v1  300 
v1  400 
v2  500 
v2  600 
v2  700 
v3  800 
v3  900 

i Datenrahmen in neuen Datenrahmen DF2 zu verwandeln suchen:

id val1 val2 
v1  100  200 
v1  200  300 
v1  300  400 
v1  400  100 
v2  500  600 
v2  600  700 
v2  700  500 
v3  800  900 
v3  900  800 

also den nächsten Mal in Folge Wert in der nächsten Spalte verschieben und auch eine Art von Gruppierung von IDs,

ich versuchte mit df.shift(), aber nicht funktioniert.

Gibt es eine Alternative dazu?

Answer 1

Wir wollen np.roll verwenden, um die Aufgabe innerhalb von Gruppen zu erreichen. Durch die Verwendung von transform umgehen wir das Auflegen des internen Gruppenindex.

df.groupby('id').value.transform(np.roll, shift=-1) 

0 200 
1 300 
2 400 
3 100 
4 600 
5 700 
6 500 
7 900 
8 800 
Name: value, dtype: int64 

---

Wir können eine neue Spalte mit einer Kopie von df mit assign

df.assign(val2=df.groupby('id').value.transform(np.roll, shift=-1)) 

    id value val2 
0 v1 100 200 
1 v1 200 300 
2 v1 300 400 
3 v1 400 100 
4 v2 500 600 
5 v2 600 700 
6 v2 700 500 
7 v3 800 900 
8 v3 900 800 

hinzufügen oder eine neue Spalte direkt an df anstelle

df['val2'] = df.groupby('id').value.transform(np.roll, shift=-1) 

df 

    id value val2 
0 v1 100 200 
1 v1 200 300 
2 v1 300 400 
3 v1 400 100 
4 v2 500 600 
5 v2 600 700 
6 v2 700 500 
7 v3 800 900 
8 v3 900 800 

Answer 2

Ich glaube, Sie brauchen numpy.roll:

df['val2'] = df.groupby('id')['value'].apply(lambda x: pd.Series(np.roll(x, -1))).values 
print (df) 
    id value val2 
0 1 100 200 
1 1 200 300 
2 1 300 400 
3 1 400 100 
4 2 500 600 
5 2 600 700 
6 2 700 500 
7 3 800 900 
8 3 900 800