SuperAPI AI 工具集
合作ENAI 行业资讯收藏
Skill 广场pymatgen

pymatgen

数据与分析

查看项目

更新于 10-27 12:33

技能介绍

用途

pymatgen（Python Materials Genomics）是用于计算材料科学的 Python 库。本资料是一份系统性使用指南，帮助用户分析晶体结构、访问材料数据库，并与计算化学程序对接。

主要能力

指南按任务类型组织，覆盖以下方向：

  • 结构创建与加载：从零创建结构、从文件（CIF、POSCAR、XYZ）读取、从 Materials Project 获取、由对称性生成。
  • 结构分析：对称性与空间群分析、结构比较、性质计算、结构可视化。
  • 结构操作：创建超胞与 slab、原子替换、应用对称操作、为 NEB 做插值。
  • 材料数据库：查询 Materials Project、获取结构与性质、构建相图。
  • 电子结构：解析能带结构、绘制 DOS、分析 VASP 输出。
  • 输入生成：创建 VASP 输入、生成 Gaussian 输入、设置计算。

使用方式

基本工作流模式：

# 1. 导入模块
from pymatgen.core import Structure, Lattice, Element

# 2. 加载或创建结构
structure = Structure.from_file("POSCAR")

# 3. 分析或操作
print(f"Formula: {structure.composition.reduced_formula}")
print(f"Space group: {structure.get_space_group_info()}")

# 4. 按需变换
supercell = structure * (2, 2, 1)  # 2x2x1 超胞

# 5. 写出结果
supercell.to(filename="POSCAR_supercell")

常见任务速查：

  • 加载结构：Structure.from_file("file.cif")，自动识别格式。
  • 创建结构：参见 references/core-objects.md 中的 Element、Lattice、Structure 创建方法。
  • 分析对称性：SpacegroupAnalyzer(struct).get_space_group_symbol()。
  • 生成超胞：structure * (2, 2, 1) 或使用 transformations。
  • 查询 Materials Project：MPRester(key).get_structure_by_material_id("mp-149")。
  • 生成 VASP 输入：MPRelaxSet(struct).write_input("dir")。
  • 绘制能带/DOS：参见 references/electronic-structure.md。

任务路由

  • 核心对象与创建 → references/core-objects.md：从零创建结构、理解 Element/Site/Structure 类、使用 Lattice 对象、创建分子、成分分析。关键类包括 Element、Species、Lattice、Site、PeriodicSite、Structure、Molecule、Composition。
  • 文件输入输出 → references/file-io.md：读取 CIF、POSCAR、XYZ 文件，写出结构，格式转换，解析计算输出。支持 CIF、POSCAR/CONTCAR（VASP）、XYZ、JSON 及多种计算化学程序格式。
  • 结构分析 → references/structure-analysis.md：查找空间群、对称操作、结构比较、获取原胞/惯用胞、近邻分析。关键工具包括 SpacegroupAnalyzer、StructureMatcher、VoronoiAnalysis、距离计算。
  • 结构变换 → references/transformations.md：创建超胞、制作 slab/表面、替换元素、扰动结构、高通量工作流。关键模块包括 pymatgen.transformations.standard_transformations、pymatgen.transformations.advanced_transformations、pymatgen.alchemy。
  • Materials Project API → references/materials-project.md：查询材料数据库、按化学式获取结构、获取计算性质、访问实验数据、构建化学体系。关键类为 MPRester，需要 materialsproject.org 的 API 密钥。
  • 相图 → references/phase-diagrams.md：构建相图、分析稳定性、寻找分解产物、Pourbaix 图、巨势图。关键类包括 PhaseDiagram、PhaseDiagramError、PourbaixDiagram。
  • 电子结构 → references/electronic-structure.md：分析能带结构、绘制 DOS、寻找带隙、分析轨道贡献、电子性质计算。关键模块包括 pymatgen.electronic_structure.bandstructure、pymatgen.electronic_structure.dos、pymatgen.electronic_structure.plotter。
  • VASP 集成 → references/vasp-integration.md：生成 VASP 输入、解析 VASP 输出、创建输入集、高通量 VASP、自定义 INCAR 设置。关键类包括 MPRelaxSet、MPStaticSet、Vasprun、Outcar 解析器、Poscar、Incar。

常见模式

  • 模式 1：结构分析——加载 → 分析成分/对称性 → 获取原胞/惯用胞。完整工作流见 examples/structure_analysis.py。
  • 模式 2：从 Materials Project 到计算——查询 MP → 获取结构 → 生成 VASP 输入 → 自定义设置。见 examples/mp_to_vasp.py。
  • 模式 3：高通量替换——加载基础结构 → 应用变换 → 写出结果。见 examples/substitution_study.py。
  • 模式 4：电子结构分析——解析 vasprun.xml → 提取能带结构/DOS → 绘图与分析。见 examples/band_structure.py。

安装与配置

安装 pymatgen：

pip install pymatgen
# 或安装可选依赖
pip install pymatgen[all]

配置 Materials Project API：

# 在 materialsproject.org 注册获取 API 密钥
pmg config --add PMG_MAPI_KEY your_api_key_here

设置 VASP 赝势路径：

pmg config --add PMG_VASP_PSP_DIR /path/to/vasp/potentials

配置文件位于 ~/.pmgrc.yaml，示例内容：

PMG_MAPI_KEY: your_api_key_here
PMG_VASP_PSP_DIR: /path/to/vasp/potentials
PMG_DEFAULT_FUNCTIONAL: PBE

常见问题与解决

  • 结构未定义：出现 NameError: name 'Structure' is not defined，解决方法是 from pymatgen.core import Structure。
  • API 密钥无效：MPRester 返回认证错误，需从 materialsproject.org 获取 API 密钥（免费账户），用 pmg config --add PMG_MAPI_KEY your_key 配置，或直接传入 MPRester("your_key")。
  • POTCAR 生成失败：无法写出 POTCAR 文件，需设置赝势目录 pmg config --add PMG_VASP_PSP_DIR /path，确保 VASP 赝势已正确安装，并检查目录结构符合预期格式。
  • 可选依赖导入错误：绘图或专用模块出现 ImportError，可执行 pip install pymatgen[all]，或单独安装 pip install matplotlib（绘图）、pip install scipy（分析工具）。

最佳实践

  • 面向对象：使用 Structure/Molecule 对象而非原始坐标，利用内置方法（成分、对称性等），链式操作以保持清晰。
  • 序列化：使用 as_dict() / from_dict() 持久化，优先用 JSON 而非 pickle，复杂对象使用 MontyEncoder/MontyDecoder。
  • 变换：使用 Transformation 类以保证可复现性，TransformedStructure 记录历史，Alchemy 框架用于高通量。
  • 输入集：标准计算使用 MPRelaxSet、MPStaticSet，创建后修改 Incar 进行自定义，自定义工作流使用 InputGenerator。
  • Materials Project：使用上下文管理器 with MPRester(...) as mpr:，尽量批量查询，缓存结果以避免重复 API 调用。

示例与参考文档

examples/ 目录包含完整工作流：structure_analysis.py（结构分析）、mp_query.py（Materials Project 查询）、phase_diagram.py（相图构建）、band_structure.py（电子结构分析）、vasp_workflow.py（VASP 计算设置）、substitution_study.py（高通量替换）。

references/ 目录包含：core-objects.md（Element、Structure、Lattice、Composition）、file-io.md（各文件格式读写）、structure-analysis.md（对称性、比较、近邻）、transformations.md（超胞、替换、扰动）、materials-project.md（API 使用与查询）、phase-diagrams.md（相图构建）、electronic-structure.md（能带与 DOS）、vasp-integration.md（VASP 输入输出处理）、properties.md（计算性质）、visualization.md（结构可视化）。

外部资源

  • 官方文档：https://pymatgen.org/
  • Materials Project：https://materialsproject.org/
  • GitHub：https://github.com/materialsproject/pymatgen
  • 论坛：https://matsci.org/pymatgen

限制

本资料未提供 pymatgen 的收费、价格、平台支持范围、性能指标或效果数据，这些信息暂未提供。