able to switch

Author: xinrong-meng

python/pyspark/pandas/data_type_ops/num_ops.py

@@ -248,16 +248,21 @@ def truediv(self, left: IndexOpsLike, right: Any) -> SeriesOrIndex:
         if not is_valid_operand_for_numeric_arithmetic(right):
             raise TypeError("True division can not be applied to given types.")
 
-        right = transform_boolean_operand_to_numeric(right, spark_type=left.spark.data_type)
-
         def truediv(left: PySparkColumn, right: Any) -> PySparkColumn:
-            return F.when(
-                right == 0,
-                F.when(left < 0, F.lit(float("-inf")))
-                .when(left > 0, F.lit(float("inf")))
-                .otherwise(F.lit(np.nan)),
-            ).otherwise(left / right)
+            if not get_option("compute.ansi_mode_support"):
+                return F.when(
+                    F.lit(right != 0) | F.lit(right).isNull(),
+                    left.__div__(right),
+                ).otherwise(F.lit(np.inf).__div__(left))
+            else:
+                return F.when(
+                    right == 0,
+                    F.when(left < 0, F.lit(float("-inf")))
+                    .when(left > 0, F.lit(float("inf")))
+                    .otherwise(F.lit(np.nan)),
+                ).otherwise(left / right)
 
+        right = transform_boolean_operand_to_numeric(right, spark_type=left.spark.data_type)
         return numpy_column_op(truediv)(left, right)
 
     def floordiv(self, left: IndexOpsLike, right: Any) -> SeriesOrIndex:
@@ -337,12 +342,22 @@ def truediv(self, left: IndexOpsLike, right: Any) -> SeriesOrIndex:
             raise TypeError("True division can not be applied to given types.")
 
         def truediv(left: PySparkColumn, right: Any) -> PySparkColumn:
-            return F.when(
-                right == 0,
-                F.when(left < 0, F.lit(float("-inf")))
-                .when(left > 0, F.lit(float("inf")))
-                .otherwise(F.lit(np.nan)),
-            ).otherwise(left / right)
+            if not get_option("compute.ansi_mode_support"):
+                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)
+                    )
+                )
+            else:
+                return F.when(
+                    right == 0,
+                    F.when(left < 0, F.lit(float("-inf")))
+                    .when(left > 0, F.lit(float("inf")))
+                    .otherwise(F.lit(np.nan)),
+                ).otherwise(left / right)
 
         right = transform_boolean_operand_to_numeric(right, spark_type=left.spark.data_type)
         return numpy_column_op(truediv)(left, right)