Записывать длинные строки в файл без разрывов строк в python - PullRequest
Новая
       35

Записывать длинные строки в файл без разрывов строк в python

0 голосов
/ 09 февраля 2020

У меня проблемы с печатью длинных строк в файл с использованием Python. В частности, я использую следующий код для вывода coords, который представляет собой массив numpy (10 на 2). 

with open('MD_traj.yml', 'a+', newline='') as outfile:
     outfile.write('# Output data of MD simulation\n') 
     outfile.write('x-coordinates: ' + str(coords[:, 0]) + '\n')
     outfile.write('y-coordinates: ' + str(coords[:, 1]) + '\n')

Что мне нужно в выходном файле: 

x-coordinates: [ 1.31142392 -1.10193486 -0.66411767 -0.98806056 -0.38443227 -0.99041216 0.99185667 -0.20955044 -0.17442841  1.43698767]
y-coordinates: [-1.2635609   0.50664106  1.0458195  -1.16822174  0.46595609  1.1952824 -0.87070535  0.4427565  -0.79005599  0.74077841]

Однако в моем выходном файле строки были разбиты на две части, что усложнило синтаксический анализ файла. Как показано ниже. 

x-coordinates: [ 1.31142392 -1.10193486 -0.66411767 -0.98806056 -0.38443227 -0.99041216
  0.99185667 -0.20955044 -0.17442841  1.43698767]
y-coordinates: [-1.2635609   0.50664106  1.0458195  -1.16822174  0.46595609  1.19528241
 -0.87070535  0.4427565  -0.79005599  0.74077841]

Может ли кто-нибудь предложить свои предложения по этому вопросу? Я потратил много времени на поиски решений, но не повезло. Заранее большое спасибо!

1 Ответ

0 голосов
/ 09 февраля 2020

Возможно, все дело в реализации __str__ для NumPy.

Вот мой код. Это решает вашу проблему. Просто используйте метод tolist() из NumPy объекта coords. 

import numpy as np
from random import random


x = [random() for _ in range(10)]
y = [random() for _ in range(10)]
coords = np.vstack([x, y])
with open('data.yml', 'a+', newline='') as outfile:
    coords_list = coords.tolist()
    outfile.write('# Output data of MD simulation\n')
    outfile.write('x-coordinates: ' + str(coords_list[0]) + '\n')
    outfile.write('y-coordinates: ' + str(coords_list[1]) + '\n')

Результат 

# Output data of MD simulation
x-coordinates: [0.8686902412164521, 0.478781961466336, 0.6641005825531633, 0.39111314403044306, 0.9438645478501313, 0.8371483392442387, 0.675984748690976, 0.7254844588305968, 0.7879460984009438, 0.7033985196947845]
y-coordinates: [0.8587330241195635, 0.4748353213357631, 0.20692421648029558, 0.8039948888725431, 0.9731648049162153, 0.7237063173464939, 0.8089361624221216, 0.16435387677097268, 0.944345230621302, 0.2901067965594649]
...