Самый эффективный способ конвертировать многомерный список Python в файл CSV? - PullRequest
0 голосов
/ 15 октября 2019

Я хочу вывести многомерный список в файл CSV.

В настоящее время я создаю новый объект DataFrame и преобразовываю его в CSV. Я знаю о модуле csv, но я не могу понять, как это сделать без ручного ввода. Метод populate позволяет пользователю выбирать, сколько строк и столбцов они хотят. По сути, переменная данных обычно будет иметь форму [[x1, y1, z1], [x2, y2, z2], ...]. Любая помощь приветствуется.

FROM populator IMPORT populate
FROM pandas IMPORT DataFrame

data = populate()
df = DataFrame(data)
df.to_csv('output.csv')

1 Ответ

0 голосов
/ 15 октября 2019

CSV - это не что иное, как строки, разделенные запятыми, для каждого столбца и строки, разделенные для каждой строки, что вы можете сделать следующим образом:

data = [[1, 2, 4], ['A', 'AB', 2], ['P', 23, 4]]

data_string = '\n'.join([', '.join(map(str, row)) for row in data])
with open('data.csv', 'wb') as f:
   f.write(data_string.encode())
...