TensorFlow读取后缀名为csv的文件中的smiles号并画成结构式批量打印到屏幕。
时间: 2024-04-10 09:28:40 浏览: 114
TensorFlow 读取CSV数据的实例
您可以使用TensorFlow的csv模块来读取后缀名为csv的文件中的smiles号,并使用RDKit库将其转换为分子结构式,并批量打印到屏幕上。下面是一个示例代码:
```python
import tensorflow as tf
import rdkit.Chem as Chem
# 读取csv文件
dataset = tf.data.experimental.CsvDataset('your_file.csv', [tf.string], header=True)
# 遍历数据并打印分子结构式
for data in dataset:
smiles = data[0].numpy().decode() # 将字符串转换为numpy数组并解码为字符串
mol = Chem.MolFromSmiles(smiles) # 转换为分子对象
if mol is not None:
Chem.Draw.MolToMPL(mol) # 使用RDKit绘制分子结构
```
请确保安装了TensorFlow和RDKit库,并将'your_file.csv'替换为您的csv文件路径。这段代码将逐行读取csv文件中的smiles号并将其转换为分子结构,并使用RDKit库绘制在屏幕上。
阅读全文