如何移动 pandas DataFrame 中的多行？

Question

我有以下 pandas 数据框：

import pandas as pd
data = {'one' : pd.Series([1.], index=['a']), 'two' : pd.Series([1., 2.], index=['a', 'b']), 'three' : pd.Series([1., 2., 3., 4.], index=['a', 'b', 'c', 'd'])}
df = pd.DataFrame(data)
df = df[["one", "two", "three"]]


   one  two  three
a  1.0  1.0    1.0
b  NaN  2.0    2.0
c  NaN  NaN    3.0
d  NaN  NaN    4.0

我知道如何按列移动元素 upwards/downwards，例如

df.two = df.two.shift(-1)

   one  two  three
a  1.0  2.0    1.0
b  NaN  NaN    2.0
c  NaN  NaN    3.0
d  NaN  NaN    4.0

但是，我想将行 a 中的所有元素移动到两列，并将行 b 中的所有元素移动到一列。最终数据框如下所示：

   one  two  three
a  NaN  NaN    1.0
b  NaN  NaN    2.0
c  NaN  NaN    3.0
d  NaN  NaN    4.0

如何在 pandas 中做到这一点？

Answer 1

您可以转置初始 DF，这样您就可以将行标签作为列名来访问，以便执行 shift 操作。

将相应列的内容向下移动这些数量，然后重新转置以获得所需的结果。

df_t = df.T
df_t.assign(a=df_t['a'].shift(2), b=df_t['b'].shift(1)).T

Answer 2

恕我直言，这是一种更像 Python 的方式：

df.loc['a'] = df.loc['a'].shift(periods=2,axis=0)

注意在这种情况下 .loc returns 一个系列，所以 axis=0 是可选的。

如何移动 pandas DataFrame 中的多行？

How to shift several rows in a pandas DataFrame?

python

shift

dataframe

pandas