código leet 004: Mediana de dos matrices ordenadas

Método general: encuentre el valor K-ésimo más grande en dos matrices

def find_kMax(array01,array02,k):
    km1=max(array01)
    km2=max(array02)
    for i in range(1,k):
        while km1 in array01:
            array01.remove(km1)
        while km2 in array02:
            array02.remove(km2)
        km1=max(array01)
        km2=max(array02)
    print(km1,km2)
arr01=[1,2,3,4,5,6,6,7,8,9]
arr02=[0,1,3,5,7,8,9,11,55,33]
find_kMax(arr01,arr02,2)