Question

我有以下数据框：

frame=pd.DataFrame({"col1":[1,5,9,4,7,3],"col2":[5,8,7,9,3,4],"col3":[3,4,2,7,9,1],
          "col4":[2,4,7,4,9,0],"col5":[3,4,5,2,1,1],"col6":[8,7,5,4,1,2]})

它产生以下输出：

    col1    col2    col3    col4    col5    col6
0     1       5      3       2        3       8
1     5       8      4       4        4       7
2     9       7      2       7        5       5
3     4       9      7       4        2       4
4     7       3      9       9        1       1
5     3       4      1       0        1       2

我想创建一个新的数据框，使col1和col2，col3和col4以及col5和col6不同

预期的输出是这样的：

    col1-col2   col3-col4   col5-col6
0      -4           1           -5
1      -3           0           -3
2       2          -5            0
3      -5           3           -2
4       4           0            0
5      -1           1           -1

预先感谢

Answer 1

dfr = pd.DataFrame({'col1-col2': frame.col1 - frame.col2,
                    'col3-col4': frame.col3 - frame.col4,
                    'col5-col6': frame.col5 - frame.col6})

Answer 2

如果许多列使用常规解决方案-选择成对和不成对的列，则转换为numpy数组，并由构造函数创建新的DataFrame：

#pandas 0.24+
arr = frame.iloc[:, ::2].to_numpy() - frame.iloc[:, 1::2].to_numpy()
#pandas below
#arr = frame.iloc[:, ::2].values - frame.iloc[:, 1::2].values

c = [f'{a}-{b}' for a, b in zip(frame.columns[::2], frame.columns[1::2])]
df = pd.DataFrame(arr, columns=c)
print (df)
   col1-col2  col3-col4  col5-col6
0         -4          1         -5
1         -3          0         -3
2          2         -5          0
3         -5          3         -2
4          4          0          0
5         -1          1         -1

如果性能很重要，请先转换为numpy数组，存储为变量，然后进行索引：

#pandas 0.24+
arr = frame.to_numpy()
#pandas below
#arr = frame.values
c = [f'{a}-{b}' for a, b in zip(frame.columns[::2], frame.columns[1::2])]
df = pd.DataFrame(arr[:, ::2] - arr[:, 1::2], columns=c)

Answer 3

df = pd.DataFrame(frame.apply(lambda x: [x['col1']-x['col2'],x['col3']-x['col4'],x['col5']-x['col6']],axis=1).tolist())
df.rename({0:'col1-col2',1:'col3-col4',2:'col4-col5'},axis=1)

   col1-col2  col3-col4  col4-col5
0         -4          1         -5
1         -3          0         -3
2          2         -5          0
3         -5          3         -2
4          4          0          0
5         -1          1         -1

在熊猫中逐列减去元素

3 个答案: