Python量子力学计算模拟以及数据可视化

简介:

前言

本文为作者对其开源项目VASPy的说明文章。VASPy是一个纯Python编写的处理VASP文件数据以及进行数据快速可视化的库,基于OOP的思想提供了操作VASP文件的友好的接口,可以帮助使用者快速编写处理VASP相关文件的脚本,以提升效率。VASPy的项目仍处于起步阶段,希望大家可以都贡献出自己的力量使其壮大起来。

VASP简介

对于广大做计算化学或者材料模拟的同学肯定听说过VASP的大名或者其科学研究与其息息相关。 
VASP的全称是Vienna Ab-initio Simulation Package,是维也纳大学Hafner课题组开发的进行电子结构计算和量子力学-分子动力学模拟的软件包,目前是材料模拟和计算物质科学研究中最流行的商业软件之一。关于VASP的详细介绍可以参见其官方主页


( http://www.vasp.at/)
VASPy项目简介

VASPy的思想是将VASP相关的文件都视为可操作的对象,通过友好的接口对一个或者多个VASP对象进行快速的操作以提升工作效率。目前已兼容Python2 和 Python3。

VASPy的GitHub地址: https://github.com/PytLab/VASPy
VASPy的PyPI地址: https://pypi.python.org/pypi/vaspy/
使用说明



安装

VASPy库已上传至PyPI可以通过pip来进行安装:

$ pip install vaspy

从源码安装:

$ git clone git@github.com:PytLab/VASPy.git
$ cd vaspy
$ python setup.py install

VASPy包的文件结构

VASPy/

├── LICENSE

├── MANIFEST

├── MANIFEST.in

├── README.rst

├── requirements.txt

├── scripts

│   ├── change_incar_parameters.py

│   ├── create_inputs.py

│   └── ...

├── setup.cfg

├── setup.py

├── tests

│   ├── incar_test.py

│   ├── __init__.py

│   ├── oszicar_test.py

│   ├── outcar_test.py

│   ├── testdata

│   │   ├── CONTCAR

│   │   ├── DOS_SUM

│   │   ├── ELFCAR

│   │   └── ...

│   └── ...

└── vaspy

    ├── __init__.py

    ├── iter.py

    ├── matstudio.py

    └── ...

文件操作举例

目前VASPy提供了操作INCAR、POSCAR、OUTCAR、XDATCAR、ELFCAR等的接口,这里对其中的部分进行简要的举例介绍。

操作INCAR文件

INCAR是VASP做电子结构计算的参数设置文件,VASPy提供了InCar类可以方便获取INCAR文件的信息以及进行自定义的修改并生成新的INCAR文件。

VASPy/

├── LICENSE

├── MANIFEST

├── MANIFEST.in

├── README.rst

├── requirements.txt

├── scripts

│   ├── change_incar_parameters.py

│   ├── create_inputs.py

│   └── ...

├── setup.cfg

├── setup.py

├── tests

│   ├── incar_test.py

│   ├── __init__.py

│   ├── oszicar_test.py

│   ├── outcar_test.py

│   ├── testdata

│   │   ├── CONTCAR

│   │   ├── DOS_SUM

│   │   ├── ELFCAR

│   │   └── ...

│   └── ...

└── vaspy

    ├── __init__.py

    ├── iter.py

    ├── matstudio.py

    └── ...

通过此类操作便可以快速写出批量修改INCAR文件的脚本,附上代码链接(https://github.com/PytLab/VASPy/blob/master/scripts/change_incar_parameters.py)

操作POSCAR/CONTCAR/XDATCAR等含有结构坐标的文件

操作结构文件可以获取相应结构的信息,例如晶胞参数、晶胞体积等。

In [7]: from vaspy.atomco import PosCar



In [8]: poscar = PosCar("POSCAR")



In [9]: poscar.bases

Out[9]: 

array([[  7.29321435,  -4.21073927,   0.        ],

       [  0.        ,   8.42147853,   0.        ],

       [ -0.        ,   0.        ,  16.87610843]])



In [10]: poscar.get_volume()

Out[10]: 1036.5246404472209



In [11]: poscar.data

Out[11]: 

array([[ 0.24466667,  0.224     ,  0.13581544],

       [ 0.02244444,  0.11288889,  0.27163089],

       [ 0.13355555,  0.00177777,  0.        ],

       ...                                 

同时结构坐标类中还提供了三维空间坐标转换接口,例如Cartisan坐标与Direct坐标的相互转换。

In [7]: from vaspy.atomco import PosCar



In [8]: poscar = PosCar("POSCAR")



In [9]: poscar.bases

Out[9]: 

array([[  7.29321435,  -4.21073927,   0.        ],

       [  0.        ,   8.42147853,   0.        ],

       [ -0.        ,   0.        ,  16.87610843]])



In [10]: poscar.get_volume()

Out[10]: 1036.5246404472209



In [11]: poscar.data

Out[11]: 

array([[ 0.24466667,  0.224     ,  0.13581544],

       [ 0.02244444,  0.11288889,  0.27163089],

       [ 0.13355555,  0.00177777,  0.        ],

       ...                                 

从XDATCAR中获取迭代的结构信息。

from vaspy.atomco import XdatCar

>>> xdatcar = XdatCar("XDATCAR")

>>> for step, data in xdatcar:

>>>     print(step)

>>>     print(xdatcar.dir2cart(xdatcar.bases, data))

操作OUTCAR文件

OUTCAR是VASP最重要的输出文件,我们可以从中获取计算过程中基本上所有的信息。

获取迭代过程中原子的受力信息:

In [4]: from vaspy.iter import OutCar



In [5]: outcar = OutCar("OUTCAR_freq", poscar='POSCAR_freq')



In [9]: outcar.forces()  # 最近一次迭代中结构中原子在各个方向上的受力

Out[9]: 

([[2.79563, 0.85618, 1.19698],

  [4.47844, 0.86375, 4.78817],

  [2.37243, -0.5474, 3.59093],

  [3.91022, -0.54961, 7.26487],

  ...                        ])

如果要获取所以迭代步中的受力信息,需要使用OutCar提供的受力信息迭代器:

for forces in outcar.force_iterator:

    # Do something with forces tuple.

    ...

OutCar类对于含有频率计算的信息的文件会做频率收取操作,可以方便获取频率相关数据:

In [16]: outcar.freq_info

Out[16]: ('index', 'freq_type', 'THz', '2PiTHz', 'cm-1', 'meV', 'coordinates', 'deltas')



In [17]: outcar.freq_types

Out[17]: [['f', 'f', 'f'], ['f', 'f', 'f/i']]



In [19]: outcar.zpe

Out[19]: 0.1117761635



In [20]: for freq_info in outcar.freq_iterator:

    ...:     # Do something with frequency data

    ...:     ...

对于其他文件的操作这里就不进行一一介绍了。

VASP数据可视化

可视化分割后的DOS(态密度)数据

可视化的过程中可以选择进行d-band center的计算并显示。

In [1]: from vaspy.electro import DosX



In [2]: dos = DosX('DOS_SUM')



In [3]: dos.plotsum(0, (5, 10))

效果图:


ELFCAR/ CHGCAR数据的可视化

电荷数据主要是通过对三维矩阵进行处理后进行绘制,可以选择surface以及二维map和标量场的显示模式。

In [1]: from vaspy.electro import ElfCar



In [2]: elfcar = ElfCar("ELFCAR")



In [3]: elfcar.plot_contour()

3D 等值线图, 这需要安装Mayavi模块来进行绘制。
In [4]: elfcar.plot_contour3d()


绘制标量场,同样需要Mayavi的支持。
In [5]: elfcar.plot_field()

CHGCAR也是Fortran顺序的三维矩阵,绘制道理相同,因此可以用继承自ElfCar的ChgCar类来进行CHGCAR相关的绘制,例如差分电荷图。

In [4]: from vaspy.electro import ChgCar


In [5]: chgcar = ChgCar("CHGCAR_diff")


In [6]: chgcar.plot_contour()

操作MaterialStudio中的xsd以及xtd等文件的接口

VASPy还提供了一个方便将Material Studio中的xsd文件与VASP文件互通的接口,通过VASPy中的XsdFileXtdFile类可以抽取文件中的晶格结构信息并结合VASP相关的类进行VASP文件的创建,同样可以方便的讲VASP的文件生成相应的用Material Studio可以显示的文件包括讲XDATCAR生成相应的*.arc*.xtd来显示动画效果。

附上脚本的链接,此脚本就是利用VASPy的接口将Material Studio文件和VASP的文件进行相互转换。

由MaterialStudio的xsd文件生成VASP输入文件的脚本
由VASP的输出文件生成相应的MaterialStudio可以显示的xsd文件的脚本
由MaterialStudio的轨迹文件生成VASP进行NEB搜索过渡态的输入文件的脚本

由VASP结果生成MaterialStudio的轨迹文件的动画效果图:


结语

VASPy最初的想法是通过Python优雅简洁的特点将VASP的文件处理进行模块化,从而省去了重复写脚本的所花费的精力,使操作VASP文件像操作变量一样简单有效。 
目前本项目都是在作者工作需要的基础上不断对其功能和接口进行完善,但仍只是冰山一角,希望做计算模拟使用VASP的Pythoner们能不断参与进来,使其更加出色和高效。


原文发布时间为:2016-12-30

本文作者:邵正将

本文来自云栖社区合作伙伴“Python中文社区”,了解相关信息可以关注“Python中文社区”微信公众号

相关文章
|
22天前
|
机器学习/深度学习 数据可视化 搜索推荐
基于python的汽车数据可视化、推荐及预测系统
本研究围绕汽车数据可视化、推荐及预测系统展开,结合大数据与人工智能技术,旨在提升用户体验与市场竞争力。内容涵盖研究背景、意义、相关技术如 Python、ECharts、协同过滤及随机森林回归等,探讨如何挖掘汽车数据价值,实现个性化推荐与智能预测,为汽车行业智能化发展提供支持。
|
1月前
|
机器学习/深度学习 新能源 调度
电力系统短期负荷预测(Python代码+数据+详细文章讲解)
电力系统短期负荷预测(Python代码+数据+详细文章讲解)
171 1
|
1月前
|
缓存 API 网络架构
淘宝item_search_similar - 搜索相似的商品API接口,用python返回数据
淘宝联盟开放平台中,可通过“物料优选接口”(taobao.tbk.dg.optimus.material)实现“搜索相似商品”功能。该接口支持根据商品 ID 获取相似推荐商品,并返回商品信息、价格、优惠等数据,适用于商品推荐、比价等场景。本文提供基于 Python 的实现示例,包含接口调用、数据解析及结果展示。使用时需配置淘宝联盟的 appkey、appsecret 和 adzone_id,并注意接口调用频率限制和使用规范。
|
24天前
|
存储 监控 API
Python实战:跨平台电商数据聚合系统的技术实现
本文介绍如何通过标准化API调用协议,实现淘宝、京东、拼多多等电商平台的商品数据自动化采集、清洗与存储。内容涵盖技术架构设计、Python代码示例及高阶应用(如价格监控系统),提供可直接落地的技术方案,帮助开发者解决多平台数据同步难题。
|
27天前
|
存储 JSON 算法
Python集合:高效处理无序唯一数据的利器
Python集合是一种高效的数据结构,具备自动去重、快速成员检测和无序性等特点,适用于数据去重、集合运算和性能优化等场景。本文通过实例详解其用法与技巧。
81 0
|
2月前
|
存储 Web App开发 前端开发
Python + Requests库爬取动态Ajax分页数据
Python + Requests库爬取动态Ajax分页数据
|
2月前
|
JSON API 数据格式
Python采集京东商品评论API接口示例,json数据返回
下面是一个使用Python采集京东商品评论的完整示例,包括API请求、JSON数据解析
|
9天前
|
JSON API 数据安全/隐私保护
Python采集淘宝评论API接口及JSON数据返回全流程指南
Python采集淘宝评论API接口及JSON数据返回全流程指南
|
11天前
|
数据采集 数据可视化 关系型数据库
基于python大数据的电影数据可视化分析系统
电影分析与可视化平台顺应电影产业数字化趋势,整合大数据处理、人工智能与Web技术,实现电影数据的采集、分析与可视化展示。平台支持票房、评分、观众行为等多维度分析,助力行业洞察与决策,同时提供互动界面,增强观众对电影文化的理解。技术上依托Python、MySQL、Flask、HTML等构建,融合数据采集与AI分析,提升电影行业的数据应用能力。
|
26天前
|
数据采集 搜索推荐 数据可视化
基于python大数据的商品数据可视化及推荐系统
本系统基于Python、Django与ECharts,构建大数据商品可视化及推荐平台。通过爬虫获取商品数据,利用可视化技术呈现销售趋势与用户行为,结合机器学习实现个性化推荐,助力电商精准营销与用户体验提升。

推荐镜像

更多