Use python code to generate the lammps data file for the nanosphere model of the grafted polymer chain

下面是用 Python 代码生成 nanosphere 模型的 LAMMPS 数据文件的示例：

import random

# 定义 nanosphere 的半径
radius = 10

# 定义聚合物链的长度
chain_length = 20

# 定义聚合物链的起始位置
chain_start = (0, 0, 0)

# 定义聚合物链每个单位的极性
polarity = 1

# 定义聚合物链的总质量
total_mass = 100

# 定义每个聚合物单位的质量
unit_mass = total_mass / chain_length

# 定义 LAMMPS 数据文件的头部
header = """
LAMMPS Data File for Nanosphere Grafted Polymer Chain

{} atoms
{} bonds

1 atom types
1 bond types

0 {} xlo xhi
0 {} ylo yhi
0 {} zlo zhi

Masses

1 {}

Atoms

""".format(chain_length + 1, chain_length, radius*2, radius*2, radius*2, unit_mass)

# 定义聚合物单位的初始位置
positions = [chain_start]

# 随机生成聚合物链的位置
for i in range(chain_length):
  x = positions[-1][0] + random.uniform(-1, 1)
  y = positions[-1][1] + random.uniform(-1, 1)
  z = positions[-1][2] + random.uniform(-1, 1)
  positions.append((x, y, z))

# 生成 LAMMPS 数据文件的原子部分
atoms = ""
for i, pos in enumerate(positions):
  atoms += "{} 1 {} {} {} {}\n".format(i+1, pos[0], pos[1], pos[2], polarity)

# 生成 LAMMPS 数据文件的键部分
bonds = ""
for i in range(chain_length):
  bonds += "{} 1 {} {}\n".format(i+1, i+1, i+2)

# 将头部、原子部分、键部分拼接在一起，得到完整的 LAMMPS 数据文件
lammps_data = header +