Экспорт списка массива и списка в CSV-файл в Python - PullRequest
Новая
       6

Экспорт списка массива и списка в CSV-файл в Python

0 голосов
/ 04 февраля 2019

Я нашел несколько ссылок, которые помогли пользователям писать списки в CSV-файл на Python.Однако я не смог найти решение для конкретного формата списка, с которым я работаю.У меня есть список, в котором каждый индекс имеет следующую форму: [array([x1,x2,x3,x4]),[x5,x6]], а в моем списке тысячи индексов.Я хочу экспортировать этот список в CSV-файл, где x1 - первый столбец, x2 - второй, а x6 - шестой столбец.Я попробовал библиотеку панд следующим образом.Тем не менее, это сбрасывает массив в первом столбце и список в другой столбец, предоставляя, таким образом, только два столбца, а не шесть, которые я хочу.

td_df = pd.DataFrame(training_data_arr)
td_df.to_csv('DataSet.csv', header=False, index=False)

Более того, я не хочу, чтобы мои данные перезаписывались каждый раз, когда я запускаю код.Скорее, я хочу, чтобы мои данные записывались из следующей пустой строки в CSV-файле

РЕДАКТИРОВАТЬ: мои данные выглядят как 

[[[0.004, 0.336, 0.009, -0.212],[1,0]], [[0.24, 0.66, 0.09, -0.2],[1,0]], [[-0.11, 0.16, 0.77, -0.2],[0,1]]...]

1 Ответ

0 голосов
/ 04 февраля 2019

Чтобы добавить файл CSV, откройте файл самостоятельно и передайте его в to_csv, например: 

with open('CartPoleDataSet.csv', 'a') as f:
    td_df.to_csv(f, header=False, index=False)

Похоже, вы просто хотите сгладить данные перед их экспортом.

Один из способов сгладить это: 

In []
pd.DataFrame(td_df.sum(axis=1).values.tolist())

Out[]:
      0     1     2      3  4  5
0 0.004 0.336 0.009 -0.212  1  0
1 0.240 0.660 0.090 -0.200  1  0
...