Панды читают фрейм данных с пустым столбцом массива - PullRequest
0 голосов
/ 02 июля 2018

У меня есть функция, которая выполняет некоторую обработку данных на фрейме данных pandas. Один из столбцов в фрейме данных - это пустой массив. Я сохраняю фрейм данных в csv, затем, когда я пытаюсь его прочитать, он читает массив как строку. Я пытаюсь понять, как пройти через это.

После обработки данных я записываю свой фрейм данных в виде csv:

df.to_csv('./dataframe')

В другом блокноте я прочитаю этот кадр данных из памяти:

df = pd.read_csv('./dataframe')

Затем я пытаюсь извлечь столбец с массивом numpy и объединить их в один большой двумерный массив:

npArray = np.zeros((df.shape[0], 4000)) #each numpy array 
                                          #in the dataframe has 4000 elements
i = 0
for vector in train:
    npArray[i] = vector[0]
    i += 1

Когда я пытаюсь это сделать, я получаю следующую ошибку:

ValueError: could not convert string to float: '['

Это происходит потому, что после того, как я прочитал файл обратно, он читает его как строку, а не как пустой массив. Я попытался добавить аргумент dtype в функцию read_csv и указать np.array, но это также дало мне ошибку.

Спасибо за любую помощь.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...