Python coding for direct method