将多列值更改为二进制值答案

【问题标题】：Changing multiple column values to binary values将多列值更改为二进制值
【发布时间】：2018-08-19 03:39:11
【问题描述】：

我以前问过这个问题，但我得到的答案并不像我想象的那样完美，所以我在这里。

上一个问题：Defining a function for changing column values and creating new datasets

我正在尝试定义一个函数，它将采用数据框并更改列中的值以创建多个新数据框。

例如，从 df1 看起来像：

  df1:

  class    colB    colC
0   1      1b      1c
1   2      2b      2c
2   3      3b      3c
3   1      4b      4c
4   2      5b      5c

我正在尝试创建多个二进制类来实现一对多分类。所以函数会创建...

df2:
  class    colB    colC
0   1      1b      1c
1   -1      2b      2c
2   -1      3b      3c
3   1      4b      4c
4   -1      5b      5c

df3:
  class    colB    colC
0   -1      1b      1c
1   1      2b      2c
2   -1      3b      3c
3   -1      4b      4c
4   1      5b      5c

df4:
  class    colB    colC
0   -1      1b      1c
1   -1      2b      2c
2    1      3b      3c
3   -1      4b      4c
4   -1      5b      5c

等等。所有唯一值都是从 1 到 120 的增量值。

上一个答案给出的问题 (np.identity) 是它创建的数据帧将每个单个值都视为 1 或 -1，而不是相应地将相同的值归类为同一类。

谢谢

【问题讨论】：

介意仔细检查df4 的输入吗？我认为只有第二行应该是1

标签： python pandas function dataframe classification

【解决方案1】：

使用np.where 和unique 的类似想法（再次重命名class 列，使其不会覆盖内置名称）：

dfs = [
    df1.assign(class_=np.where(df1['class_'].eq(i), 1, -1)) for i in df1['class_'].unique()
]

for d in dfs:
    print(d, end='\n\n')

   class_ colB colC
0       1   1b   1c
1      -1   2b   2c
2      -1   3b   3c
3       1   4b   4c
4      -1   5b   5c

   class_ colB colC
0      -1   1b   1c
1       1   2b   2c
2      -1   3b   3c
3      -1   4b   4c
4       1   5b   5c

   class_ colB colC
0      -1   1b   1c
1      -1   2b   2c
2       1   3b   3c
3      -1   4b   4c
4      -1   5b   5c

【讨论】：

你很棒。谢谢大家：D

【解决方案2】：

与@user3483203 类似，但使用mask 和fillna：

[df.assign(**{'class' : df['class'].mask(df['class'].ne(cls)).fillna(-1)}) 
     for cls in df['class'].unique()
]

[   class colB colC
 0    1.0   1b   1c
 1   -1.0   2b   2c
 2   -1.0   3b   3c
 3    1.0   4b   4c
 4   -1.0   5b   5c,    class colB colC
 0   -1.0   1b   1c
 1    2.0   2b   2c
 2   -1.0   3b   3c
 3   -1.0   4b   4c
 4    2.0   5b   5c,    class colB colC
 0   -1.0   1b   1c
 1   -1.0   2b   2c
 2    3.0   3b   3c
 3   -1.0   4b   4c
 4   -1.0   5b   5c]

【讨论】：