import pymatgen
from mp_api.client import MPRester
from pymatgen.io.cif import CifWriter
API_KEY = ""#注册账号时给的KEY,具体位置是官网右上角my DashBoard
mpr = MPRester(API_KEY)#创建MPRester对象进行认证
docs = mpr.summary.search(band_gap=value,total_magnetization=value)#这里设定搜索条件
id_list = []
for doc in docs:#获取材料索引号
id_list.append(doc.material_id)
for id in id_list:#读取材料的晶体结构并且下载
structure = mpr.get_structure_by_material_id(id)
cif_writer = CifWriter(structure)
cif_writer.write_file("SemiDatasets/"+id+".cif")
第一次用python下载材料数据集,用的新版API,由于没看文档摸索了很久
文档如下: