python 查找数组中的素数（质数）

def find_prime(nums):
 result=[]
 for num in nums:
    if num > 1:
      for i in range(2,num):
          if (num % i) == 0:
              break
      else:
          result.append(num)
    else:
      pass
 return result
find_prime([3,5,11,12])