python - 在 Pandas DataFrame 中从另一行中减去一行

Question

我正在尝试从 Pandas DataFrame 中的另一行中减去一行。我在一个数字列之前有多个描述符列，这迫使我在两个描述符列上设置 DataFrame 的索引。

当我这样做时，无论 set_index() 列列表中列出的第一列名称是什么，我都会得到一个 KeyError。在这种情况下，它是“COL_A”:

df = pd.DataFrame({'COL_A': ['A', 'A'], 
                   'COL_B': ['B', 'B'], 
                   'COL_C': [4, 2]})
df.set_index(['COL_A', 'COL_B'], inplace=True)
df.iloc[1] = (df.iloc[1] / df.iloc[0])
df.reset_index(inplace=True)


KeyError: 'COL_A'

我没有再考虑这个问题，无法弄清楚为什么 KeyError 是这样解决的。

Answer 1

我想快速回答这个问题。这就是我最终的解决方案。

>>> df = pd.DataFrame(data=[[5,5,5,5], [3,3,3,3]], index=['r1', 'r2'])
>>> df
    0  1  2  3
r1  5  5  5  5
r2  3  3  3  3
>>> df.loc['r3'] = df.loc['r1'] - df.loc['r2']
>>> df
    0  1  2  3
r1  5  5  5  5
r2  3  3  3  3
r3  2  2  2  2
>>>