狗血的pandas where函数 - mask 函数和 where 作用刚好相反

这几天在看“hands-on machine learning with sklearn and tensorflow” 的第二章节,狗血的事情发生了。其中的一部分是需要将median_income划分为5个等分,然后大于5的部分全部归类为5,代码如下:

    housing["income_cat"] = np.ceil(housing["median_income"] / 1.5)
    housing["income_cat"].where(housing["income_cat"] < 5, 5.0, inplace=True)

结果我看了大半天还以为housing["income_cat"] < 5这里写错了,后来才发现mask 函数和 where 作用刚好相反。

  s = pd.Series(range(5))
  s.where(s > 1, 10)
  0    10.0
  1    10.0
  2    2.0
  3    3.0
  4    4.0

  s.mask(s > 1, 10)
  0    0.0
  1    1.0
  2    10.0
  3    10.0
  4    10.0
  df = pd.DataFrame(np.arange(10).reshape(-1, 2), columns=['A', 'B'])
  m = df % 3 == 0
  # df.where(m, np.array([1,2,3,4,5]).reshape(-1, 5))  #此句话报错
  df.where(m, -df)
  A  B
  0  0 -1
  1 -2  3
  2 -4 -5
  3  6 -7
  4 -8  9

--------------------- 
作者:依斐 
来源:CSDN 
原文:https://blog.csdn.net/dss_dssssd/article/details/82818587 
版权声明:本文为博主原创文章,转载请附上博文链接!
添加新评论