📚 课程目录第4章 Python光学环境搭建

4. Python光学环境搭建：安装numpy、scipy、matplotlib、ray-tracing库

好，咱们开始动手了。这一章说白了就是给电脑“喂”光学计算的能力。很多初学者一上来就急着写透镜代码，结果import就报错——嗯，我当年也干过这事。

Python本身只是个空壳子，真正干活的是这些第三方库。咱们要装的四个库，各有各的脾气。我按依赖顺序来，你跟着走就行。

4.1 环境准备：先搞定Python和pip

在装库之前，先确认两件事：Python版本和pip是否正常。我个人习惯用Python 3.8到3.11之间的版本，太新或太旧都可能跟ray-tracing库打架。

打开终端（Windows用cmd或PowerShell，Mac/Linux用终端），敲下面这行：

python --version
pip --version

如果看到版本号，说明环境就绪。如果提示“不是内部或外部命令”，那得先去python.org下载安装，记得勾选“Add Python to PATH”。

注意： 我曾经遇到过pip指向了错误的Python版本，结果库装到了另一个环境里，import死活找不到。建议用 python -m pip install 来确保安装到当前Python。

4.2 核心计算库：numpy + scipy

这两个是光学计算的基石。numpy管矩阵运算，scipy管数值优化——TIR透镜的曲面拟合全靠它们。

安装命令很简单：

pip install numpy scipy

装完后，我习惯先做个快速验证：

python -c "import numpy; print(numpy.__version__); import scipy; print(scipy.__version__)"

如果输出版本号，就说明装好了。没报错就是好消息。

小技巧： 国内用户如果下载慢，可以加镜像源：pip install numpy scipy -i https://pypi.tuna.tsinghua.edu.cn/simple

4.3 可视化利器：matplotlib

做光学设计，光有数据不行，得看得见。matplotlib就是咱们的“眼睛”。TIR透镜的轮廓曲线、配光曲线、照度分布图，全得靠它画。

安装：

pip install matplotlib

验证代码：

python -c "import matplotlib; print(matplotlib.__version__)"

我个人习惯在Jupyter Notebook里用matplotlib，交互式调参特别方便。如果你也想用，可以顺带装个Jupyter：

pip install jupyter

4.4 光学专用库：ray-tracing

这个库是咱们课程的核心。它专门做几何光学追迹，能定义透镜、光源、探测器，然后算光线路径。说实话，市面上纯Python的光学追迹库不多，ray-tracing算是比较好上手的。

安装：

pip install ray-tracing

踩坑提醒： 这个库的包名是 ray-tracing（带连字符），但import的时候要用下划线：import ray_tracing。我第一次装完就卡在这儿了，找了半天才发现是导入名的问题。

验证安装：

python -c "import ray_tracing; print('OK')"

如果输出“OK”，恭喜你，环境搭好了。

4.5 一键安装脚本

嫌一条条敲太麻烦？我写了个批处理脚本，你直接跑就行：

# Windows (save as setup_env.bat)
pip install numpy scipy matplotlib ray-tracing
python -c "import numpy; import scipy; import matplotlib; import ray_tracing; print('All libraries installed successfully!')"
pause

# Mac/Linux (save as setup_env.sh)
#!/bin/bash
pip install numpy scipy matplotlib ray-tracing
python -c "import numpy; import scipy; import matplotlib; import ray_tracing; print('All libraries installed successfully!')"

4.6 知识体系总览

下面这张图把四个库的关系和分工说清楚了：

4.7 常见问题速查表

问题现象	可能原因	解决办法
pip命令找不到	Python未加入PATH	重装Python，勾选“Add to PATH”
安装速度极慢	默认源在国外	加 -i 参数用国内镜像
import ray_tracing报错	装成了ray-tracing但导入名不对	用 `import ray_tracing`（下划线）
numpy版本冲突	多个Python环境混用	用虚拟环境隔离
matplotlib不显示图形	缺少GUI后端	装 `pip install pyqt5` 或换Jupyter

我的建议： 用虚拟环境来管理项目依赖。每个光学项目单独一个环境，互不干扰。命令很简单：

python -m venv tir_env
tir_env\Scripts\activate  # Windows
source tir_env/bin/activate  # Mac/Linux

激活后再装库，所有东西都隔离在这个环境里。做完项目直接删文件夹，干干净净。