Python基础之函数定义及文件修改
函数
函数的定义
#登录函数和注册函数
def register():
"""注册函数"""
username = input('请输入你的用户名:')
password = input('请输入你的密码:')
#按照“用户名:密码|”的格式存储
with open('test.txt', 'w', encoding='utf8') as fw:
fw.write(f'{username}:{password}|')
def login():
"""登录函数"""
username = input('请输入你的用户名:')
password = input('请输入你的密码:')
#从文件中读数据进行匹配
with open('test.txt', 'r', encoding='utf8') as fr:
data = fr.read()
info_list = data.split('|')
info_user = f'{username}:{password}'
if info_user in info_list:
print('登录成功')
else:
print('登陆失败')
#直接用以下语句进行调用
#register()
#login()
#注释写在函数名的下一行未来忘记函数功能时,可用print(login.__doc__)进行查询
三种定义方式及其调用
#无参函数
def add():
"""无参函数"""
y = input('num1:')
x = input('num2:')
return int(x) + int(y)
res = add(10, 30)
print(res * 12)
#有参函数
def add(x, y)
"""有参函数"""
return int(x)+int(y)
print(add(10, 20))
#空函数
def func():
pass
##函数返回值
def add(x, y)
return x, y, x + y #return可以返回任意数据类型,不加括号默认返回元组类型
x,y,z = add(1, 2) # 可以用解压缩
print(x, y, z)
文件修改
思路:对一个文件中的数据进行修改,通常是先对该文件进行读操作,然后修改文件内容,把修改的内容写入新的文件中,在把旧文件remove掉并重命名新文件。
import os #导入os库,用于对文件进行修改和删除操作
with open('test.py', 'r', encoding='utf8') as fr, \
open('test_swap.py', 'w', encoding='utf8') as fw:
for i in fr:
data = i.replace('123', '456) #将文件中的123修改为456
fw.write(data)
os.remove('test.py') #移除原文件
os.rename('test_swap.py', 'test.py') #重命名新文件