Форма массива после извлечения из Pandas Dataframe - PullRequest
0 голосов
/ 15 октября 2018

У меня есть столбец в Dataframe, где каждая ячейка имеет массивный массив размером (300,).

Когда я извлекаю значения этого столбца с помощью метода .values, я получаю пустой массив формы (N,), где N - количество строк в кадре данных.И каждый элемент N является массивом (300,).Я ожидал, что извлеченная форма будет (Nx300).

Так что я бы хотел, чтобы извлеченная колонка была (Nx300).Я попытался использовать pd.as_matrix(), но это все равно дает мне массив данных (N,).

Есть предложения?

1 Ответ

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

Вы можете использовать numpy.concatenate, преобразовать в список и привести к array:

a = np.random.randint(10, size=300)
print (a.shape)
(300,)

df = pd.DataFrame({ 'A':[a,a,a]})

arr = np.array(np.concatenate(df.values).tolist())
print (arr.shape)
(3, 300)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...