数组按照奇数偶数输出

面试中遇到的手写题目：
例如给出数组A：[1,3,2,4,2,5,6,9,7,8]
要求输出：[1,2,3,4,5,2,9,6,7,8]

def sortArrayByParityII(A):
        A_len = len(A)
        i = 0
        j = 1
        while i < A_len:
            if A[i] % 2 ==0:
                while A[j] %2 == 0:
                    j+=2
                A[i],A[j] = A[j],A[i]
            i+=2

        return A

if __name__ == '__main__':
    A = [1,3,2,4,2,5,6,9,7,8]
    b=sortArrayByParityII(A)
    print(b)