Сбор данных Aws —

Question

У меня есть требование: у меня есть 3 столбца A,B,C в данных, которые я хочу поставить if..else/условие переключения в столбцах C, и если это правда, то вернуть значения A*B. Это означает вывод любых математических функций результата условия в столбцах A,B. Пожалуйста, помогите мне, как добиться этого в AWS-DataBrew. Пример... A,B,C Столбцы входных данных..

Ожидается результат ниже в Databrew – A,B,C – столбцы входных данных.

      If(C.equals('XYZ'))
{
   new Column D = A*B;
}else if(C.equals('ABC'))
{new Column D = A/B; }
else {new Column D = A}