当前位置：首页 > article >正文

python加载chgcar, aeccar压缩数据

article 2024/9/20 23:31:25

数据来源：materials project网站下载的.json.gz压缩包
使用gzip包可以免解压直接读取
解压出来json文件的数据结构大概如代码所示
使用pymatgen.io.vasp的Chgcar对象，可以装载从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)
``