代码工具箱分享：10 个日常使用脚本，解决编程高频问题

2023-12-01 18:12:29

提升工作效率：10 个 Python 脚本，解决常见编程问题

身为一名程序员，我们每天都会遇到各种编程问题。虽然 Python 的标准库可以轻松解决一些问题，但其他问题却需要我们编写自己的脚本。以下 10 个日常使用脚本可以帮助我们有效解决这些高频问题。

1. 测网速，选择最佳服务器

import speedtest

def test_speed():
  """测试网速并返回结果."""

  speedtest = Speedtest()
  speedtest.get_servers()
  best_server = speedtest.get_best_server()
  download_speed = speedtest.download()
  upload_speed = speedtest.upload()

  return best_server, download_speed, upload_speed


if __name__ == "__main__":
  best_server, download_speed, upload_speed = test_speed()
  print(f"最佳服务器：{best_server}")
  print(f"下载速度：{download_speed} Mbps")
  print(f"上传速度：{upload_speed} Mbps")

这个脚本使用 Speedtest 库来测试网速并确定最佳服务器，以便优化网络性能。

2. 格式化 JSON 数据

import json

def format_json(json_string):
  """格式化 JSON 字符串."""

  return json.dumps(json.loads(json_string), indent=2)


if __name__ == "__main__":
  json_string = '{"name": "John Doe", "age": 30, "city": "New York"}'
  formatted_json = format_json(json_string)
  print(formatted_json)

这个脚本使用 json 库来格式化 JSON 字符串，使其更易于阅读和理解。

3. 比较两个列表

def compare_lists(list1, list2):
  """比较两个列表并返回差异."""

  diff = set(list1) ^ set(list2)
  return list(diff)


if __name__ == "__main__":
  list1 = [1, 2, 3, 4, 5]
  list2 = [2, 3, 4, 5, 6]
  diff = compare_lists(list1, list2)
  print(diff)

这个脚本使用集合运算来比较两个列表并返回它们的差异，这对于检测两个数据集之间的差异非常有用。

4. 转换进制

def convert_base(number, base):
  """将数字转换为指定进制."""

  if base < 2 or base > 36:
    raise ValueError("Base must be between 2 and 36.")

  digits = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  result = ""

  while number > 0:
    result = digits[number % base] + result
    number //= base

  return result


if __name__ == "__main__":
  number = 100
  base = 2
  result = convert_base(number, base)
  print(result)

这个脚本将数字转换为指定进制，这对于在不同进制系统之间转换数据非常有用。

5. 随机生成字符串

import random

def generate_random_string(length):
  """生成指定长度的随机字符串."""

  chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
  random_string = ''.join(random.choice(chars) for i in range(length))
  return random_string


if __name__ == "__main__":
  length = 10
  random_string = generate_random_string(length)
  print(random_string)

这个脚本使用随机模块来生成指定长度的随机字符串，这对于生成密码、令牌或其他需要唯一性的数据非常有用。

6. 计时器

import time

class Timer:
  """计时器."""

  def __init__(self):
    self.start_time = time.time()

  def stop(self):
    """停止计时器并返回运行时间."""

    self.end_time = time.time()
    return self.end_time - self.start_time


if __name__ == "__main__":
  timer = Timer()
  time.sleep(5)
  elapsed_time = timer.stop()
  print(f"运行时间：{elapsed_time} 秒")

这个脚本创建一个计时器类，可以跟踪和测量代码的执行时间，这对于优化性能和识别瓶颈非常有用。

7. 文件操作

import os

def read_file(filename):
  """读取文件并返回文件内容."""

  with open(filename, "r") as f:
    file_content = f.read()

  return file_content


def write_file(filename, data):
  """将数据写入文件."""

  with open(filename, "w") as f:
    f.write(data)


if __name__ == "__main__":
  filename = "test.txt"
  data = "Hello, world!"
  write_file(filename, data)
  file_content = read_file(filename)
  print(file_content)

这个脚本提供了读取和写入文件的基本功能，这对于与文件系统交互和处理文本数据非常有用。

8. 创建目录

import os

def create_directory(directory_path):
  """创建目录."""

  os.makedirs(directory_path, exist_ok=True)


if __name__ == "__main__":
  directory_path = "my_directory"
  create_directory(directory_path)

这个脚本使用 os 模块来创建目录，这对于组织文件和保持文件系统井然有序非常有用。

9. 删除目录

import os

def delete_directory(directory_path):
  """删除目录."""

  os.rmdir(directory_path)


if __name__ == "__main__":
  directory_path = "my_directory"
  delete_directory(directory_path)

这个脚本使用 os 模块来删除目录，这对于清理文件系统和释放磁盘空间非常有用。

10. 解压压缩文件

import zipfile

def unzip_file(zip_file_path, destination_path):
  """解压压缩文件."""

  with zipfile.ZipFile(zip_file_path, 'r') as zip_file:
    zip_file.extractall(destination_path)


if __name__ == "__main__":
  zip_file_path = "my_file.zip"
  destination_path = "destination_path"
  unzip_file(zip_file_path, destination_path)