python - 如何根据特定列在 Pandas Dataframe 中向上移动行

Question

如何在不影响其他列顺序的情况下，将某一特定列的一行中的所有值向上移动？

例如，假设我有以下代码:

import pandas as pd
data= {'ColA':["A","B","C"],
        'ColB':[0,1,2],
        'ColC':["First","Second","Third"]}

df = pd.DataFrame(data)

print(df)

我会看到以下输出:

  ColA  ColB    ColC
0    A     0   First
1    B     1  Second
2    C     2   Third

在我的例子中，我想验证 B 列没有任何 0，如果有，它被删除，它下面的所有其他值被推高，其他列的顺序不受影响。据推测，我会看到以下内容:

  ColA  ColB    ColC
0    A   1     First
1    B   2    Second
2    C   NaN   Third

我不知道如何使用 drop() 或 shift() 方法来做到这一点。

谢谢

Answer 1

让我们做简单的排序

invalid=0
df['ColX']=sorted(df.ColB,key=lambda x : x==invalid)
df.ColX=df.ColX.mask(df.ColX==invalid)
df
Out[351]: 
  ColA  ColB    ColC  ColX
0    A     0   First   1.0
1    B     1  Second   2.0
2    C     2   Third   NaN