在 Python 中，使用 WebSocket 进行通信是一种实现客户端与服务器之间双向、实时数据交换的有效方式。WebSocket 协议允许在单个持久连接上进行全双工通信，这意味着客户端和服务器可以同时发送和接收数据。 ### 安装 WebSocket 库 要开始使用 WebSocket，首先需要安装一个支持 WebSocket 的 Python 库。`websockets` 是一个广泛使用的库，可以通过 `pip` 安装： ```bash pip install websockets ``` ### 实现 WebSocket 客户端 一旦安装了 `websockets` 库，就可以开始编写客户端代码来连接 WebSocket 服务器并与其通信。下面是一个简单的 WebSocket 客户端示例： ```python import asyncio import websockets async def connect_to_server(): async with websockets.connect("ws://localhost:8765") as websocket: # 向服务器发送消息 await websocket.send("Hello, Server!") # 接收来自服务器的消息 response = await websocket.recv() print(f"Received from server: {response}") # 运行客户端 asyncio.get_event_loop().run_until_complete(connect_to_server()) ``` ### 实现 WebSocket 服务器 同样地，可以编写一个简单的 WebSocket 服务器来监听客户端连接，并与之通信： ```python import asyncio import websockets async def echo(websocket, path): async for message in websocket: print(f"Received from client: {message}") # 将收到的消息回传给客户端 await websocket.send(f"Echo: {message}") # 启动服务器 start_server = websockets.serve(echo, "localhost", 8765) asyncio.get_event_loop().run_until_complete(start_server) asyncio.get_event_loop().run_forever() ``` ### 使用 WebSocket 进行实时数据可视化 WebSocket 不仅可以用来发送简单的文本消息，还可以用来传输实时数据，并且可以与图形库如 Matplotlib 结合使用以实现数据可视化。以下是一个实时接收数据并绘图的例子： ```python import asyncio import websockets import matplotlib.pyplot as plt # 初始化绘图 plt.ion() fig, ax = plt.subplots() async def data_client(): async with websockets.connect("wss://dataserver.com/data") as websocket: print("已连接到数据服务器") x = [] y = [] while True: message = await websocket.recv() value = float(message) x.append(len(x) + 1) y.append(value) # 更新绘图 ax.plot(x, y, 'b-') plt.draw() plt.pause(0.001) # 运行数据客户端 asyncio.get_event_loop().run_until_complete(data_client()) ``` 以上代码展示了如何创建一个 WebSocket 客户端，该客户端连接到提供实时数据的服务器，并使用 Matplotlib 实时绘制接收到的数据[^5]。