【数组与循环】寻找素数（Python）（简单）

描述
输出n以内所有的素数。

保证 n 是100以内的整数。

样例
样例 1:

输入：5
输出：[2, 3, 5]

class Solution:
    """
    @param n: an integer
    @return: return all prime numbers within n.
    """
    def prime(self, n):
        # write your code here
        if n < 2:
            return []
        elif n == 2:
            return [2]
        elif n == 3:
            return [2,3]
        ans = [2,3]
        for i in range(4,n+1):
            label = False
            for j in range(2,i):
                if i % j == 0:
                    label = True
            if label == False:
                ans.append(i)
        return ans