# coding=utf-8 #加载化学库 from rdkit import Chem from rdkit.Chem import Draw from rdkit.Chem import AllChem import pandas as pd # 读取 CSV 文件 data = pd.read_csv('dataSetA.csv') # 提取 rxn_smiles 列 rxn_smiles = data['rxn_Smiles'] # 遍历每个 rxn_smiles 字符串并打印 for smi in rxn_smiles: print(smi) rxn = Chem.AllChem.ReactionFromSmarts(smi) if rxn is not None: # 绘制反应结构 img = Draw.ReactionToImage(rxn) img.show() else: print("Failed to parse rxn_smiles.")最后要保存成图片要怎么写

要将绘制的反应结构保存为图片，您可以使用`img.save`方法来保存图片。以下是修改后的代码示例：

# 遍历每个 rxn_smiles 字符串并保存图片
for i, smi in enumerate(rxn_smiles):
    print(smi)
    rxn = Chem.AllChem.ReactionFromSmarts(smi)
    if rxn is not None:
        # 绘制反应结构
        img = Draw.ReactionToImage(rxn)
        img.save(f'reaction_{i}.png')
    else:
        print("Failed to parse rxn_smiles.")

在上述代码中，添加了一个`enumerate`函数来获得循环的索引值，以便在保存文件时为每个反应结构创建唯一的文件名。通过在`img.save`方法中传递文件名参数，可以将绘制的反应结构保存为PNG文件。

请注意，将文件名设置为`reaction_{i}.png`会根据循环的索引值创建唯一的文件名。如果您希望使用其他文件名格式，请根据需要进行修改。确保在适当的位置进行替换，以指定您希望保存图像的文件路径和文件名。