вычислить значение на основе некоторых условий и присвоить его новой переменной

Question

Я вычисляю значение чистого баланса по условию и хочу сохранить его в новой переменной. Новая переменная должна хранить это вычисленное значение как целое число или число с плавающей точкой, а не как массив

Я пробовал следующий код:

  #variable = something if condition else something_else
  mar_final_bal = x_start_bal+df2['credit_line']+df2['Net_Balance'] if 
  df2['month' == 'March-2016']
  apr_final_bal = mar_final_bal+df2['credit_line']+df2['Net_Balance'] if 
  df2['month' == 'Apr-2016']

mar_final_bal и apr_final_bal - две мои переменные, которые я хочусоздать используя условия на правой стороне

Answer 1

Очевидно, что вы новичок в использовании Панд. Синтаксис выглядит более псевдоподобным, чем код панд. IIUC, это то, что вы имели в виду:

mar_final_bal = x_start_bal+df2.loc[df2['month'] == 'March-2016', 'credit_line'].sum() + df2.loc[df2['month'] == 'March-2016', 'Net_Balance'].sum()
apr_final_bal = mar_final_bal+df2.loc[df2['month'] == 'Apr-2016', 'credit_line'].sum() + df2.loc[df2['month'] == 'Apr-2016', 'Net_Balance'].sum()