python - 交换某些行中的列

Question

我有这样的 DataFrame:

     A            B      C
"010-1"       "Car"     500
"010-2"     "Train"     302
"Result"      "010"      23
"011-1"      "Ship"     321
"011-2"     "Plane"     321
"011-3"      "Bike"     321
"Result"      "011"     321

现在可以在 "Result" 列 A< 中的行中移动 A 和 B 列 使用 Pandas ？

所以最终结果是:

     A            B       C
"010-1"       "Car"     500
"010-2"     "Train"     302
  "010"     "Result"     23
"011-1"      "Ship"     321
"011-2"     "Plane"     321
"011-3"      "Bike"     321
  "011"    "Result"     321

编辑:实际上，除了 A 和 B 之外，可能还有不止一列

Answer 1

你可以使用 numpy 函数 where():

原始数据框:

In[6]: df
Out[6]: 
        A      B    C
0   010-1    Car  500
1   010-2  Train  302
2  Result    010   23
3   011-1   Ship  321
4   011-2  Plane  321
5   011-3   Bike  321
6  Result    011  321

---

命令:
(使用常见的 Python 模式 x, y = y, x 来切换值)

In[7]: cond = df.A == "Result"
In[8]: df.A, df.B = np.where(cond, df.B, df.A), np.where(cond, df.A, df.B)

---

结果:

In[9]: df
Out[9]: 
       A       B    C
0  010-1     Car  500
1  010-2   Train  302
2    010  Result   23
3  011-1    Ship  321
4  011-2   Plane  321
5  011-3    Bike  321
6    011  Result  321