如何从给定数字的所有因子的向量中创建素数因子的向量？

import math

def create_prime_factors_vector(num):
    factors = []
    
    # 找到所有因子
    for i in range(1, num+1):
        if num % i == 0:
            factors.append(i)
    
    prime_factors = []
    
    # 判断每个因子是否为素数
    for factor in factors:
        is_prime = True
        for j in range(2, int(math.sqrt(factor))+1):
            if factor % j == 0:
                is_prime = False
                break
        if is_prime:
            prime_factors.append(factor)
    
    return prime_factors

# 示例调用
num = 36
prime_factors = create_prime_factors_vector(num)
print(prime_factors)