Largest Number After Digit Swaps by Parity LeetCode solution in Python

1
class Solution:
2
    def largestInteger(self, num: int):
3
        n = len(str(num))
4
        arr = [int(i) for i in str(num)]
5
        odd, even = [], []
6
        for i in arr:
7
            if i % 2 == 0:
8
                even.append(i)
9
            else:
10
                odd.append(i)
11
        odd.sort()
12
        even.sort()
13
        res = 0
14
        for i in range(n):
15
            if arr[i] % 2 == 0:
16
                res = res * 10 + even.pop()
17
            else:
18
                res = res * 10 + odd.pop()
19
        return res