Python高效工作必备：20个实用脚本推荐！

import os

def batch_rename(path, prefix):
    """为指定目录下的所有文件添加前缀"""
    for count, filename in enumerate(os.listdir(path)):
        old_path = os.path.join(path, filename)
        if os.path.isfile(old_path):
            # 分离文件名和扩展名
            name, ext = os.path.splitext(filename)
            new_name = f"{prefix}_{name}{ext}"
            new_path = os.path.join(path, new_name)
            os.rename(old_path, new_path)
    print("重命名完成！")

# 使用示例
batch_rename("./photos", "vacation")

import os

def list_all_files(directory):
    """列出目录下所有文件的完整路径"""
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            print(file_path)

# 使用示例
list_all_files("./projects")

def search_in_files(directory, keyword):
    """搜索目录下所有包含关键词的文件"""
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            try:
                with open(file_path, 'r', encoding='utf-8') as f:
                    content = f.read()
                    if keyword in content:
                        print(f"在文件中找到关键词 '{keyword}': {file_path}")
            except Exception as e:
                # 忽略无法读取的文件（如二进制文件）
                pass

# 使用示例
search_in_files("./docs", "Python脚本")

import os

def safe_write_to_file(file_path, content):
    """安全地写入文件，自动创建所需目录"""
    directory = os.path.dirname(file_path)
    if directory and not os.path.exists(directory):
        os.makedirs(directory, exist_ok=True) # exist_ok=True 防止目录已存在时报错

    with open(file_path, 'w') as f:
        f.write(content)
    print(f"文件已写入: {file_path}")

# 使用示例
safe_write_to_file("./output/logs/app.log", "这是一条日志信息。")

import csv
import json

def csv_to_json(csv_file, json_file):
    """将CSV文件转换为JSON文件"""
    data = []
    with open(csv_file, 'r', encoding='utf-8') as f:
        csv_reader = csv.DictReader(f)
        for row in csv_reader:
            data.append(row)

    with open(json_file, 'w', encoding='utf-8') as f:
        json.dump(data, f, indent=4, ensure_ascii=False)
    print(f"转换完成！JSON文件已保存至: {json_file}")

# 使用示例
csv_to_json("input.csv", "output.json")

import pandas as pd

def summarize_excel(file_path, output_file):
    """读取Excel文件，对每个工作表进行描述性统计，并保存到新文件"""
    with pd.ExcelWriter(output_file) as writer:
        # sheet_name=None 会读取所有工作表
        all_sheets = pd.read_excel(file_path, sheet_name=None)
        
        for sheet_name, df in all_sheets.items():
            # 对每个工作表进行描述性统计
            summary = df.describe(include='all')
            summary.to_excel(writer, sheet_name=f"Summary_{sheet_name}")
    print(f"汇总完成！文件已保存至: {output_file}")

# 使用示例
summarize_excel("data.xlsx", "summary_report.xlsx")

import pandas as pd

def clean_data(input_file, output_file):
    """数据清洗：填充空值并过滤掉异常值"""
    df = pd.read_csv(input_file)
    
    # 填充数字列的空值为中位数
    for col in df.select_dtypes(include=['number']).columns:
        df[col].fillna(df[col].median(), inplace=True)
    
    # 过滤掉“年龄”列中大于100的异常值
    if'age'in df.columns:
        df = df[df['age'] <= 100]
    
    df.to_csv(output_file, index=False)
    print(f"数据清洗完成！已保存至: {output_file}")

# 使用示例
clean_data("dirty_data.csv", "cleaned_data.csv")

import requests

def download_file(url, local_filename):
    """下载文件并保存到本地"""
    with requests.get(url, stream=True) as r:
        r.raise_for_status()
        with open(local_filename, 'wb') as f:
            for chunk in r.iter_content(chunk_size=8192):
                f.write(chunk)
    print(f"文件已下载: {local_filename}")

# 使用示例
download_file('https://example.com/somefile.zip', 'downloaded_file.zip')

import requests

def check_website(url):
    """检查网站状态"""
    try:
        response = requests.head(url, timeout=5)
        print(f"{url} 的状态码是: {response.status_code}")
        return response.status_code == 200
    except requests.RequestException as e:
        print(f"{url} 无法访问: {e}")
        return False

# 使用示例
check_website("https://www.google.com")

import smtplib
from email.mime.text import MIMEText
from email.header import Header

def send_email(sender, receivers, subject, content, smtp_server, password):
    """发送邮件通知"""
    message = MIMEText(content, 'plain', 'utf-8')
    message['From'] = Header(sender, 'utf-8')
    message['To'] = Header(",".join(receivers), 'utf-8')
    message['Subject'] = Header(subject, 'utf-8')

    try:
        smtp_obj = smtplib.SMTP_SSL(smtp_server) # 例如QQ邮箱：smtp.qq.com, 端口465
        smtp_obj.login(sender, password)
        smtp_obj.sendmail(sender, receivers, message.as_string())
        smtp_obj.quit()
        print("邮件发送成功")
    except smtplib.SMTPException as e:
        print(f"无法发送邮件: {e}")

# 使用示例（需配置发件人邮箱和授权码）
# send_email('sender@qq.com', ['receiver@example.com'], '脚本运行完成', '数据处理任务已成功执行！', 'smtp.qq.com', 'your_authorization_code')

from win10toast import ToastNotifier
import time

def desktop_notify(title, message, duration=5):
    """发送Windows桌面通知"""
    toaster = ToastNotifier()
    toaster.show_toast(title, message, duration=duration)
    # 等待通知结束
    time.sleep(duration)

# 使用示例
desktop_notify("脚本通知", "数据处理已完成！")

import os

def check_env_vars(*variables):
    """检查一个或多个环境变量是否存在及其值"""
    for var in variables:
        value = os.getenv(var)
        if value is None:
            print(f"环境变量 '{var}' 未设置。")
        else:
            print(f"{var} = {value}")

# 使用示例：检查JAVA_HOME和PATH
check_env_vars('JAVA_HOME', 'PATH')

import requests

def get_weather(city_name, api_key):
    """获取指定城市的天气信息（这里以OpenWeatherMap为例）"""
    base_url = "http://api.openweathermap.org/data/2.5/weather?"
    complete_url = f"{base_url}appid={api_key}&q={city_name}&units=metric&lang=zh_cn"
    
    response = requests.get(complete_url)
    data = response.json()
    
    if data["cod"] != "404":
        main = data["main"]
        weather_desc = data["weather"][0]["description"]
        temp = main["temp"]
        print(f"{city_name}的天气: {weather_desc}")
        print(f"当前温度: {temp}°C")
    else:
        print("城市未找到")

# 使用示例（需要去OpenWeatherMap申请免费API Key）
# get_weather("Beijing", "YOUR_API_KEY_HERE")

import subprocess

def run_command(command):
    """执行系统命令并打印输出"""
    try:
        # capture_output=True 捕获输出，text=True 返回字符串而非字节
        result = subprocess.run(command, shell=True, capture_output=True, text=True, timeout=30)
        print("STDOUT:")
        print(result.stdout)
        if result.stderr:
            print("STDERR:")
            print(result.stderr)
        print(f"返回码: {result.returncode}")
    except subprocess.TimeoutExpired:
        print("命令执行超时！")

# 使用示例
run_command("dir") # Windows
# run_command("ls -l") # Linux/Mac

import secrets
import string

def generate_password(length=12):
    """生成一个包含大小写字母、数字和符号的随机密码"""
    alphabet = string.ascii_letters + string.digits + "!@#$%^&*"
    password = ''.join(secrets.choice(alphabet) for i in range(length))
    return password

# 使用示例
print(f"你的新密码是: {generate_password(16)}")

import time

def timing_decorator(func):
    """一个简单的计时装饰器"""
    def wrapper(*args, **kwargs):
        start_time = time.perf_counter()
        result = func(*args, **kwargs)
        end_time = time.perf_counter()
        elapsed_time = end_time - start_time
        print(f"函数 {func.__name__} 运行耗时: {elapsed_time:.4f} 秒")
        return result
    return wrapper

# 使用示例：在要计时的函数上加上 @timing_decorator 即可
@timing_decorator
def example_function():
    time.sleep(1)

example_function()

import PyPDF2

def extract_text_from_pdf(pdf_path):
    """从PDF第一页提取文本"""
    with open(pdf_path, 'rb') as file:
        reader = PyPDF2.PdfReader(file)
        if len(reader.pages) > 0:
            first_page = reader.pages[0]
            text = first_page.extract_text()
            return text
        else:
            return"PDF为空或无法读取。"

# 使用示例
text = extract_text_from_pdf("document.pdf")
print(text)

import re

def advanced_replace(text, pattern, replacement):
    """使用正则表达式进行高级搜索替换"""
    new_text = re.sub(pattern, replacement, text)
    return new_text

# 使用示例
original_text = "我的电话是 123-4567-8901，另一个是 987-6543-210。"
# 匹配所有XXX-XXXX-XXX模式的电话号
pattern = r'\d{3}-\d{4}-\d{3,4}'
replacement = "【电话号码已隐藏】"
result = advanced_replace(original_text, pattern, replacement)
print(result)

# 这不是一个.py脚本，而是在命令行中直接运行的命令
# 在需要共享的目录下打开终端（CMD或PowerShell），运行：
python -m http.server 8000
# 然后同一局域网下的设备浏览器访问 http://你的IP地址:8000 即可

import json
from pprint import pprint

# 假设有一个复杂的嵌套字典或JSON字符串
complex_data = {'name': 'John', 'age': 30, 'pets': [{'type': 'dog', 'name': 'Spot'}, {'type': 'cat', 'name': 'Fluffy'}], 'address': {'street': '123 Main St', 'city': 'Anytown'}}

print("普通打印:")
print(complex_data) # 一行显示，难以阅读

print("\n漂亮打印:")
pprint(complex_data, indent=4, width=50) # 格式化输出，清晰易读

# 如果是JSON字符串，可以先加载
# json_data = json.loads(json_string)
# pprint(json_data)