Refactor DataFrame.sort method to use list comprehension for improved readability

Author: kosiew

python/datafusion/dataframe.py

@@ -537,10 +537,10 @@ def sort(self, *exprs: Expr | SortExpr | str) -> DataFrame:
         Returns:
             DataFrame after sorting.
         """
-        exprs_raw = []
-        for expr in exprs:
-            expr_obj = Expr.column(expr).sort() if isinstance(expr, str) else expr
-            exprs_raw.append(sort_or_default(expr_obj))
+        exprs_raw = [
+            sort_or_default(Expr.column(expr) if isinstance(expr, str) else expr)
+            for expr in exprs
+        ]
         return DataFrame(self.df.sort(*exprs_raw))
 
     def cast(self, mapping: dict[str, pa.DataType[Any]]) -> DataFrame: