[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