Matrix completion in python