如何根据python中现有列中的条件创建新列？

Question

我有一个如下所示的 dataframe：

df:

PAN_NO  COST_VALUE
AAA       -0.001          
BBB       2938080         
CCC      49224091         
DDD        100            
EEE     50236272.32

我正在尝试根据以下条件创建一个新列：

如果 df['cost_value'] >=0.001 且 df['cost_value'] <= 299985.0 那么 cost_value_group 应该是 1

如果 df['cost_value'] > 299985.0 并且 df['cost_value'] <= 2938082.40 那么 cost_value_group 应该是 2

如果 df['cost_value'] > 2938082.40 并且 df['cost_value'] <= 17399130.0 那么 cost_value_group 应该是 3

如果 df['cost_value'] > 2938082.40 并且 df['cost_value'] <= 17399130.0 那么 cost_value_group 应该是 3

如果 df['cost_value'] > 17399130.0 并且 df['cost_value'] <= 49224091.375 那么 cost_value_group 应该是 4

如果 df['cost_value'] > 49224091.375 cost_value_group 应该是 5

否则它应该是 6

EXPECTED OUTPUT:
PAN_NO  COST_VALUE COST_VALUE_Group
AAA       -0.001          1
BBB       2938080         2
CCC      49224091         5
DDD        100            1
EEE     50236272.32       6

我试着做：

def cost_value(x):
   if df['cost_value'] >= -0.001 and df['cost_value'] <= 299985.0:
   return 1
   elif df['cost_value'] > 299985.0 and df['cost_value'] <= 2938082.40:
   return 2
   elif df['cost_value'] > 2938082.40 and df['cost_value'] <= 17399130.0:
   return 3
   elif df['cost_value'] > 17399130.0 and df['cost_value'] <= 49224091.375:
   return 4
  elif df['cost_value'] > 49224091.375:
   return 5
  else:
   return 6
df['cost_value_group] = df['cost_value].apply(cost_value)

我收到一个值错误，表明系列的真实值不明确。

有人可以帮助我吗？

Answer 1

你走在正确的道路上。

尝试这个：

df['cost_value_group'] = df['cost_value'].apply(lambda x: cost_value(x))