收集了一些常用Python脚本，作为平时练手使用，也可以作为自己的笔记，用到哪个功能可以自己查询一下即可。
文件和目录管理
复制文件
import shutil
复制源文件到目标文件
shutil.copy('source.txt', 'destination.txt')
移动文件
import shutil
移动文件到新的路径
shutil.move('source.txt', 'destination.txt')
创建目录结构
import os
创建多层目录，如果已经存在则不报错
os.makedirs('dir/subdir/subsubdir', exist_ok=True)
删除空目录
import os
删除当前目录下的所有空目录
for root, dirs, files in os.walk('.', topdown=False):
for name in dirs:
dir_path = os.path.join(root, name)
if not os.listdir(dir_path):
os.rmdir(dir_path)
查找大文件
import os
查找当前目录及子目录下大于1MB的文件
for root, dirs, files in os.walk('.'):
for name in files:
if os.path.getsize(os.path.join(root, name)) > 1024 * 1024:
print(os.path.join(root, name))
检查文件是否存在
import os
检查指定文件是否存在
if os.path.exists('file.txt'):
print("File exists.")
else:
print("File does not exist.")
读取文件内容
with open('file.txt', 'r') as file:
content = file.read()
写入文件内容
with open('file.txt', 'w') as file:
file.write('Hello, World!')
批量重命名
遍历当前文件夹下的mp3文件，把000.mp3, 001.mp3更名为Question Number 1.mp3, Question Number 2.mp3
from pathlib import Path
files = Path(".").glob("*.mp3")
num = 1
for file in files:
file.replace(f"Question Number {num}.mp3")
num+=1
数据处理
读取 CSV 文件
import csv
读取 CSV 文件并打印每一行
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
写入 CSV 文件
import csv
写入数据到 CSV 文件
data = [['Name', 'Age'], ['Alice', 30], ['Bob', 25]]
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
读取 JSON 文件
import json
读取 JSON 文件
with open('data.json', 'r') as file:
data = json.load(file)
写入 JSON 文件
import json
将数据写入 JSON 文件
data = {'name': 'Alice', 'age': 30}
with open('data.json', 'w') as file:
json.dump(data, file)
过滤列表中的重复项
从列表中去除重复项
my_list = [1, 2, 2, 3, 4, 4, 5]
unique_list = list(set(my_list))
排序列表
对列表进行排序
my_list = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
sorted_list = sorted(my_list)
网络请求与爬虫
获取网页内容
import requests
发送 GET 请求并获取网页内容
response = requests.get('https://www.example.com')
print(response.text)
发送 HTTP POST 请求
import requests
发送 POST 请求并打印响应
payload = {'key1': 'value1', 'key2': 'value2'}
response = requests.post('https://httpbin.org/post', data=payload)
print(response.text)
处理 JSON 响应
import requests
获取并解析 JSON 响应
response = requests.get('https://api.example.com/data')
data = response.json()
print(data)
下载图片
import requests
下载并保存图片
img_data = requests.get('http://example.com/image.jpg').content
with open('image.jpg', 'wb') as handler:
handler.write(img_data)
自动化任务
定时执行任务
import schedule
import time
定义定时执行的任务
def job():
print("I'm working...")
每10秒执行一次任务
schedule.every(10).seconds.do(job)
while True:
schedule.run_pending()
time.sleep(1)
发送电子邮件
import smtplib
from email.mime.text import MIMEText
发送电子邮件
msg = MIMEText('Hello, this is a test email.')
msg['Subject'] = 'Test Email'
msg['From'] = 'your_email@example.com'
msg['To'] = 'recipient@example.com'
s = smtplib.SMTP('localhost')
s.send_message(msg)
s.quit()
文字处理
统计单词数
统计字符串中的单词数
text = "This is a test. This is only a test."
word_count = len(text.split())
print(f"Word count: {word_count}")
替换字符串
替换字符串中的子串
text = "Hello, World!"
new_text = text.replace("World", "Python")
print(new_text)
连接字符串
将列表中的字符串连接为一个字符串
fruits = ['apple', 'banana', 'orange']
text = ', '.join(fruits)
print(text)
格式化字符串
使用 f-string 格式化字符串
name = "Alice"
age = 30
formatted_text = f"Name: {name}, Age: {age}"
print(formatted_text)
其他常见功能
生成随机数
import random
生成1到100之间的随机整数
random_number = random.randint(1, 100)
print(random_number)
生成随机密码
python
复制代码
import random
import string
生成随机密码
password = ''.join(random.choices(string.ascii_letters + string.digits, k=12))
print(password)
读取环境变量
import os
读取指定环境变量
api_key = os.getenv('API_KEY')
print(api_key)
运行系统命令
import subprocess
运行系统命令并打印输出
result = subprocess.run(['ls', '-l'], stdout=subprocess.PIPE)
print(result.stdout.decode('utf-8'))
