Python中能否传参动态修改if比较运算符以调整矩阵排序规则？

当然可以动态修改if的判断条件，完全不用改动交换逻辑这类核心代码。下面结合你的矩阵排序需求，给出两种简单的实现方式：

方法一：利用operator模块的运算符函数 #

Python的operator模块提供了和内置运算符对应的函数，比如operator.lt对应<，operator.gt对应>，你可以把这些函数作为参数传给排序函数，动态切换判断逻辑。

先优化你原来的代码（修复全局变量、越界问题，去掉冗余的try-except），再加入动态判断的逻辑：

import operator

def sort_matrix_by_column_sum(arr, compare=operator.lt):
    col_count = len(arr[0])
    is_done = 0
    while is_done < col_count - 1:
        is_done = 0
        # 循环到倒数第二列即可，避免x+1索引越界
        for x in range(col_count - 1):
            sum_current = arr[0][x] + arr[1][x]
            sum_next = arr[0][x+1] + arr[1][x+1]
            # 用传入的compare函数替代固定运算符
            if compare(sum_current, sum_next):
                # 交换两列元素逻辑完全不变
                arr[0][x], arr[0][x+1] = arr[0][x+1], arr[0][x]
                arr[1][x], arr[1][x+1] = arr[1][x+1], arr[1][x]
            else:
                is_done += 1

使用示例：

• 降序排序（和你原逻辑一致，用<）：

  arr = [[3,1,4], [2,5,1]]
  sort_matrix_by_column_sum(arr, operator.lt)
  

• 升序排序（改用>）：

  arr = [[3,1,4], [2,5,1]]
  sort_matrix_by_column_sum(arr, operator.gt)
  

方法二：自定义lambda表达式作为判断条件 #

如果需要更灵活的判断逻辑（比如多条件组合），可以直接传入lambda表达式：

arr = [[3,1,4], [2,5,1]]
# 降序逻辑：当前列和 < 下一列和时交换
sort_matrix_by_column_sum(arr, lambda a,b: a < b)
# 升序逻辑：当前列和 > 下一列和时交换
sort_matrix_by_column_sum(arr, lambda a,b: a > b)

关键说明 #

• 核心思路是把判断条件抽象成一个可调用对象（函数或lambda），传入函数后在if语句中调用它，这样交换逻辑完全不用改动。
• 优化后的代码去掉了全局变量arr，改为参数传入，更符合Python编码规范；调整循环范围后，彻底解决了原代码中索引越界的问题，无需再用try-except兜底。

内容的提问来源于stack exchange，提问作者4lll0w_3v1l