有办法减少这种情况吗？

Question

我有这段代码，其中st是具有x和y参数的类的数组。我必须对它们的x或y进行排序。我如何减少它，所以我不必两次编写相同的代码（每种情况一次）？

def sort(st, coor):
    for i in range(len(st)):
        if coor == 'x':
            selected = st[i]
            j = i - 1
            while j >= 0 and st[j].x > selected.x:
                st[j + 1], st[j] = st[j], st[j + 1]
                j -= 1
            st[j + 1] = selected
            i += 1
        else:
            selected = st[i]
            j = i - 1
            while j >= 0 and st[j].y > selected.y:
                st[j + 1], st[j] = st[j], st[j + 1]
                j -= 1
            st[j + 1] = selected
            i += 1

Answer 1

为重复的代码提取方法。

https://refactoring.guru/extract-method

在该新方法中，与众不同的部分是调用实例的x或y属性，并且可以用对getattr的调用来代替。