Pythonで素数を表示するコード

def prime_number_array(num):
    array = []
    for i in range(1,num+1):
        tmp = []
        for c in range(1,num+1):
            if i % c == 0:
                tmp.append(i)
        
        if len(tmp) == 2:
            array.append(tmp[0])   
    return array

num = prime_number_array(10)
print(num)