在Python中，你可以编写一个函数来判断一个数是否为素数。素数是指只有1和它本身两个正因数（不包括1本身）的自然数。以下是一个简单的示例代码：

python复制代码

	def is_prime(n):
	if n <= 1:
	return False
	if n <= 3:
	return True
	if n % 2 == 0 or n % 3 == 0:
	return False
	i = 5
	while i * i <= n:
	if n % i == 0 or n % (i + 2) == 0:
	return False
	i += 6
	return True
	# 测试代码
	num = int(input("请输入一个整数: "))
	if is_prime(num):
	print(num, "是素数")
	else:
	print(num, "不是素数")

在这个代码中，is_prime 函数用于判断一个数是否为素数。首先，我们检查一些基本情况：如果 n 小于等于 1，那么它肯定不是素数；如果 n 小于等于 3，那么它是素数（因为 2 和 3 都是素数）。然后，我们检查 n 是否可以被 2 或 3 整除，如果可以，则 n 不是素数。接下来，我们使用一个循环来检查 n 是否可以被其他数整除。这个循环从 5 开始，每次增加 6（因为我们已经检查了所有形如 6k ± 1 的数）。如果 n 可以被循环中的任何数整除，那么 n 不是素数。如果循环结束而没有找到任何能整除 n 的数，那么 n 是素数。

最后，我们使用 input 函数从用户那里获取一个整数，并使用 is_prime 函数来判断这个数是否为素数，然后输出结果。