基于多个条件，使用NP.SELECT更改数据框中的值

Question

我想使用np.select更改数据框中的值 但是，我遇到了一个错误，Google没有找到针对类似问题的任何解决方案。 有人可以帮我吗？预先感谢您的时间和帮助。 这是我的数据框

import pandas as pd
import numpy as np

df = pd.DataFrame({'A':[1,0,-0.6,-1,0.7],
       'B':[-1,1,-0.3,0.5,1]})
df

，这是我的代码，其中DF值被重新编码如下：

• 值小于或等于-0.5成为-1

• 值大于-0.5，小于或等于0.5变为0

• 值大于0.5变为1

  new_df = pd.dataframe（ np.Select（[DF.LE（-0.5），df.gt（ - 。5）＆amp; df.le（0.5），df.gt（0.5）]， [-1，0，1]）， index = df.index， 列= df.columns ）

  new_df

我的代码给我以下错误，我不确定怎么了，如何解决它 keyError：＆lt;函数dispatch_to_series..column_op at 0x7f64a469d158＆gt;

Answer 1

您可以尝试尝试一下 - 我相信将小数舍入到零数字的情况下也遵守您的所有条件：

# default decimals is 0 
df_new = df.round()

# or this if you need more control on each column decimal places
df_new = df.round({'A': 0, 'B': 0})