函数部分巩固提升
练习1：实现计算求最大公约数和最小公倍数的函数。
参考答案：
def gcd(x, y):
"""求最大公约数"""
(x, y) = (y, x) if x > y else (x, y)
for factor in range(x, 0, -1):
if x % factor == 0 and y % factor == 0:
return factor
def lcm(x, y):
"""求最小公倍数"""
return x * y // gcd(x, y)练习2：实现判断一个数是不是回文数的函数。
参考答案：
def is_palindrome(num):
"""判断一个数是不是回文数"""
temp = num
total = 0
while temp > 0:
total = total * 10 + temp % 10
temp //= 10
return total == num练习3：实现判断一个数是不是素数的函数。
参考答案：
def is_prime(num):
"""判断一个数是不是素数"""
for factor in range(2, int(num ** 0.5) + 1):
if num % factor == 0:
return False
return True if num != 1 else False练习4：写一个程序判断输入的正整数是不是回文素数。
参考答案：
if __name__ == '__main__':
num = int(input('请输入正整数: '))
if is_palindrome(num) and is_prime(num):
print('%d是回文素数' % num)注意：通过上面的程序可以看出，当我们将代码中重复出现的和相对独立的功能抽取成函数后，我们可以组合使用这些函数来解决更为复杂的问题，这也是我们为什么要定义和使用函数的一个非常重要的原因。
练习5：生成验证码
要求：设计一个函数产生指定长度的验证码，验证码由大小写字母和数字构成。
参考答案：
import random
def generate_code(code_len=4):
"""
生成指定长度的验证码
:param code_len: 验证码的长度(默认4个字符)
:return: 由大小写英文字母和数字构成的随机验证码
"""
all_chars = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
last_pos = len(all_chars) - 1
code = ''
for _ in range(code_len):
index = random.randint(0, last_pos)
code += all_chars[index]
return code练习6：返回文件名后缀
设计一个函数返回给定文件名的后缀名。
参考答案：
def get_suffix(filename, has_dot=False):
"""
获取文件名的后缀名
:param filename: 文件名
:param has_dot: 返回的后缀名是否需要带点
:return: 文件的后缀名
"""
pos = filename.rfind('.')
if 0 < pos < len(filename) - 1:
index = pos if has_dot else pos + 1
return filename[index:]
else:
return ''练习7：寻找列表前两大元素
设计一个函数返回传入的列表中最大和第二大的元素的值。
参考答案：
def max2(x):
m1, m2 = (x[0], x[1]) if x[0] > x[1] else (x[1], x[0])
for index in range(2, len(x)):
if x[index] > m1:
m2 = m1
m1 = x[index]
elif x[index] > m2:
m2 = x[index]
return m1, m2练习8：打印杨辉三角。
要求：给定一个非负整数num,生成杨辉三角的前num行
在杨辉三角中，每个数是它左上方和右上方的数的和
Example:
Input: 5
Output:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]参考答案：
def main():
num = int(input('Number of rows: '))
yh = [[]] * num
for row in range(len(yh)):
yh[row] = [None] * (row + 1)
for col in range(len(yh[row])):
if col == 0 or col == row:
yh[row][col] = 1
else:
yh[row][col] = yh[row - 1][col] + yh[row - 1][col - 1]
print(yh[row][col], end='\t')
print()
if __name__ == '__main__':
main()练习9：趣味体重排序
丽在一个”减肥会所”，每到月底小丽就很担心，因为每个月都会公布一份成员体重列表，每个月她都是列表中的最后一个，这意味着她是最重的。于是她就建议会长修改规则，比如新的规则如下：
新的排序规则将是按照“权重”来进行排序，如果权重相同，则按照体重字符串来进行排序。(权重的计算规则:体重数字不同位数上的数字的总和）
给定某种正常顺序俱乐部的成员的体重字符串，如何实现按照新规则排序呢？ 测试用例: 输入: “99 189 86 75” 输出: “75 86 189 99”
解法1：
def order_weight(weight_string):
weight_list = weight_string.split(' ')
def helper(x):
if not x:
return x
y = int(x)
z = 0
while y:
z += y % 10
y = int(y / 10)
return (z, x)
weight_list.sort(key=helper)
new_weight_string = " ".join(weight_list)
return new_weight_string
if __name__ == '__main__':
weight_string = '99 189 86 75'
new_weight = order_weight(weight_string)
print(new_weight)解法2：
def order_weight(_str):
return' '.join(sorted(_str.split(' '),key = lambda x: sum(int(c) for c in x)))
if __name__ == '__main__':
weight_string = '99 189 86 75'
new_weight = order_weight(weight_string)
print(new_weight)综合案例：井字棋游戏。
import os
def print_board(board):
print(board['TL'] + '|' + board['TM'] + '|' + board['TR'])
print('-+-+-')
print(board['ML'] + '|' + board['MM'] + '|' + board['MR'])
print('-+-+-')
print(board['BL'] + '|' + board['BM'] + '|' + board['BR'])
def main():
init_board = {
'TL': ' ', 'TM': ' ', 'TR': ' ',
'ML': ' ', 'MM': ' ', 'MR': ' ',
'BL': ' ', 'BM': ' ', 'BR': ' '
}
begin = True
while begin:
curr_board = init_board.copy()
begin = False
turn = 'x'
counter = 0
os.system('clear')
print_board(curr_board)
while counter < 9:
move = input('轮到%s走棋, 请输入位置: ' % turn)
if curr_board[move] == ' ':
counter += 1
curr_board[move] = turn
if turn == 'x':
turn = 'o'
else:
turn = 'x'
os.system('clear')
print_board(curr_board)
choice = input('再玩一局?(yes|no)')
begin = choice == 'yes'
if __name__ == '__main__':
main()
