import matplotlib.pyplot as plt
import numpy as np
def write_xyz(mol_str, fname):
input_str = "$coord\n"
for i in np.arange(len(mol_str.split("\n"))):
if mol_str.split("\n")[i]:
data = mol_str.split("\n")[i].split(" ")
tmp_str = ""
for i in np.arange(len(data)):
if data[i]:
if data[i].isalpha():
save_el = data[i].lower()
else:
tmp_str += data[i] + " "
tmp_str += save_el
input_str += tmp_str + "\n"
input_str += "$end"
f = open(fname, "w")
f.write(input_str)
f.close()
return False