File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -1138,11 +1138,16 @@ def get_coordinates_xyz(filename):
1138
1138
if lines_read == n_atoms :
1139
1139
break
1140
1140
1141
- atom = re .findall (r'[a-zA-Z]+' , line )[0 ]
1142
- atom = atom .upper ()
1141
+ values = line .split ()
1143
1142
1144
- numbers = re .findall (r'[-]?\d+\.\d*(?:[Ee][-\+]\d+)?' , line )
1145
- numbers = [float (number ) for number in numbers ]
1143
+ if len (values ) < 4 :
1144
+ atom = re .findall (r'[a-zA-Z]+' , line )[0 ]
1145
+ atom = atom .upper ()
1146
+ numbers = re .findall (r'[-]?\d+\.\d*(?:[Ee][-\+]\d+)?' , line )
1147
+ numbers = [float (number ) for number in numbers ]
1148
+ else :
1149
+ atom = values [0 ].upper ()
1150
+ numbers = [float (number ) for number in values [1 :]]
1146
1151
1147
1152
# The numbers are not valid unless we obtain exacly three
1148
1153
if len (numbers ) >= 3 :
You can’t perform that action at this time.
0 commit comments