python - 合并和减去 Pandas 中的 DataFrame 列？

Question

我有一个 pandas DataFrame，类似于:

col1  col2 col3 col5
NaN    1    2    8
2     NaN   4    8
4     NaN   4    8

我想做两件事:

1) 合并第 1 列和第 2 列:

newcol1 col3 col5
1       2    8
2       4    8
4       4    8

我试过使用 .concat，但这只是连接行。我似乎无法使用带有 NaN 值的标准 + 运算符。

2) 从新的第 1 列和第 3 列中减去第 5 列，所以我最终得到:

newcol1    col3
-7         -6
-6         -4
-4         -4

试过这样做:

dataframe[['newcol1', 'col2']] - dataframe['col5']

和

dataframe[['newcol1', 'col2']].subtract(dataframe['col5'])

但两者都不起作用。

Answer 1

要获取新列，您可以使用 fillna(或 combine_first):

df['newcol1'] = df.col1.fillna(df.col2)

然后对于减法，使用 sub 并指定 axis=0 因为我们想在匹配标签时考虑行索引(而不是默认的列索引):

>>> df[['newcol1', 'col3']].sub(df['col5'], axis=0)
   newcol1  col3
0       -7    -6
1       -6    -4
2       -4    -4