def truediv(left: Column, right: Any) -> Column:
return F.when(
F.lit(right != 0) | F.lit(right).isNull(),
left.__div__(right)).otherwise(
F.when(
F.lit(left == np.inf) | F.lit(left == -np.inf),
left).otherwise(F.lit(np.inf).__div__(left)))
def floordiv(left: Column, right: Any) -> Column:
return F.when(F.lit(right is np.nan), np.nan).otherwise(
F.when(
F.lit(right != 0) | F.lit(right).isNull(),
F.floor(left.__div__(right))).otherwise(
F.when(
F.lit(left == np.inf) | F.lit(left == -np.inf),
left).otherwise(F.lit(np.inf).__div__(left))))
