python - 为列列表中的任何列是否包含值创建一个虚拟对象

Question

我对 Pandas 和 Python 比较陌生。我正在寻找有关完成以下任务的最快方法的建议。我正在尝试编写与pd.get_dummies()命令类似的代码，但将其概括为不止一列。

我想选择 dx1、dx2 和 dx3 列，并为这些列中的任何一个是否包含给定值返回一个虚拟变量。

输入数据示例：

In [2]: df
Out[2]: 
   dx1  dx2   dx3  othervar
0  1     2    3       4
1  1     3            5
2  3     4            2

输出数据：

In [2]: df
Out[2]: 
   dx1  dx2   dx3  othervar  dx_any1    dx_any2    dx_any 3   dx_any4
0  1     2    3       4        1          1           1         0
1  1     3            5        1          0           1         0
2  3     4            2        0          0           1         1  

标签： pythonpandas