您的位置:首页 > 房产 > 建筑 > python加载chgcar, aeccar压缩数据

python加载chgcar, aeccar压缩数据

2025/1/31 7:36:32 来源:https://blog.csdn.net/sc303165/article/details/142237433  浏览:    关键词:python加载chgcar, aeccar压缩数据

数据来源:materials project网站下载的.json.gz压缩包
使用gzip包可以免解压直接读取
解压出来json文件的数据结构大概如代码所示
使用pymatgen.io.vaspChgcar对象,可以装载从json文件读出的结构数据和电荷密度数据

        with gzip.open(os.path.join(self.chgcar_dir, self.chgcar_list[idx]), 'rt') as gz_file:raw_data = json.load(gz_file)struc = raw_data['data']['poscar']chgcar_data = {'total': raw_data['data']['data']["total"]["data"],'diff': raw_data['data']['data']["diff"]["data"]}chgcar_object = Chgcar(poscar=Poscar.from_dict(struc), data=chgcar_data)with gzip.open(os.path.join(self.aeccar0_dir, self.aeccar0_list[idx]), 'rt') as gz_file:raw_data0 = json.load(gz_file)with gzip.open(os.path.join(self.aeccar2_dir, self.aeccar2_list[idx]), 'rt') as gz_file:raw_data2 = json.load(gz_file)aeccar0_data = {'total': raw_data0['data']['data']["total"]["data"]}struc = raw_data0['data']['poscar']aeccar0_object = Chgcar(poscar=Poscar.from_dict(struc), data=aeccar0_data)aeccar2_data = {'total': raw_data2['data']['data']["total"]["data"]}struc = raw_data2['data']['poscar']aeccar2_object = Chgcar(poscar=Poscar.from_dict(struc), data=aeccar2_data)
``

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com