按pandas中的列名重新分类

Question

我想将测试应用于pandas数据帧，并根据测试结果在相应的数据帧中创建标记。我已经走到了这一步：

import numpy as np
import pandas as pd


matrix = pd.DataFrame({'a': [1, 11, 2, 3, 4], 'b': [5, 6, 22, 8, 9]})
flags = pd.DataFrame(np.zeros(matrix.shape), columns=matrix.columns)
flag_values = pd.Series({"a": 100, "b": 200})

flags[matrix > 10] = flag_values 

但这会引发错误

ValueError: Must specify axis=0 or 1

在这种情况下，我在哪里可以指定轴？有没有更好的方法来实现这一目标？

编辑：

我在这个例子中为“flags”寻找的结果是

a    b
0    0
100  0
0    200
0    0
0    0

Answer 1

您可以定义flags = (matrix > 10) * flag_values：

In [35]: (matrix > 10) * flag_values
Out[35]: 
     a    b
0    0    0
1  100    0
2    0  200
3    0    0
4    0    0

这取决于True having numeric value 1和False，数值为0。 在执行算术运算之前，它还依赖于Pandas基于标签的漂亮的automatic alignment DataFrames（和Series）。

Answer 2

带有mask

的

mul

flags.mask(matrix > 10,1).mul(flag_values,axis=1)

Out[566]: 
       a      b
0    0.0    0.0
1  100.0    0.0
2    0.0  200.0
3    0.0    0.0
4    0.0    0.0