Блог

Главная - Вопросы по программированию - Python: преобразовать фрейм данных с 1 столбцом в определенный ndarray

Python: преобразовать фрейм данных с 1 столбцом в определенный ndarray

#python #python-3.x #pandas #numpy #numpy-ndarray

#python #python-3.x #панды #numpy #numpy-ndarray

Вопрос:

Например, у меня есть фрейм данных pandas с 1 столбцом и n строками:

                               0
0                03110311000311
1             18003130313000313
2             36003120312000312
3             54003110311000311
4             72003100310000310
...                         ...
[1400 rows x 1 columns]

Я преобразовал этот фрейм данных в ndarray, используя функцию to_numpy(), но я получил этот ndarray:

введите описание изображения здесь

Я хотел бы получить ndarray следующим образом: y={ndarray:(1400,)}

Как это можно сделать? Спасибо!

Комментарии:

1. Используйте df.squeeze().to_numpy() или df[0].to_numpy() или df.to_numpy().squeeze()

2. Просто нарежьте массив с помощью arr[:,0] .

3. df.to_numpy() или df.values

4. @Ch3steR так просто! большое вам спасибо! Работает нормально! Можете ли вы объяснить разницу между YY ndarray и y ndarray?

Ответ №1:

Просто сначала выберите столбец:

 df['column_index'].to_numpy()

Пример:

 df = pd.DataFrame(np.arange(6))

df.to_numpy().shape            # (6,1)

df[0].to_numpy().shape       # (6,)