Início / coding / Perguntas / 77727056
Accepted
hbstha123
Asked: 2023-12-28 21:05:37 +0800 CST

Como posso usar `df.shift(n)` no dataframe do pandas de forma que eu possa trazer `n` item de baixo para cima em vez de valores `nan` ou vice-versa? [duplicado]

  • 772

Eu tenho um dataframe do pandas dfcontendo 5 linhas e 2 colunas.

A   B
0   10  0
1   20  5
2   30  10
3   40  15
4   50  20

df.to_dict()retorna

{'A': {0: 10, 1: 20, 2: 30, 3: 40, 4: 50},
 'B': {0: 0, 1: 5, 2: 10, 3: 15, 4: 20}}

Para a coluna A, quero mudar cada item para duas linhas abaixo. Em vez de ter nanvalores em cima, quero trazer dois elementos que seriam empurrados de baixo para cima.

Para a coluna B, quero fazer o oposto - mudar cada item para duas linhas acima. Em vez de ter nanvalores na parte inferior, quero trazer dois elementos que seriam empurrados de cima para baixo.

Eu posso usar df["A"].shift(2)e df["B"].shift(-2). No entanto, recebo valores nan.

Meus resultados esperados são:

A   B
0   40  10
1   50  15
2   10  20
3   20  0
4   30  5

Como posso conseguir isso?

python

1 respostas

  1. Best Answer

    Use numpy.rollem vez disso shift:

    df['A'] = np.roll(df["A"], 2)
df['B'] = np.roll(df["B"], -2)
print (df)
    A   B
0  40  10
1  50  15
2  10  20
3  20   0
4  30   5
    • 5

relate perguntas