[LintCode]Remove Element(Python)

class Solution:    """    @param A: A list of integers    @param elem: An integer    @return: The new length after remove    """    def removeElement(self, A, elem):        # write your code here        for i in range(len(A)):            if elem in A:                A.remove(elem)            else:                return A