Серия Pandas to_numpy() сохраняет имена индексов

#python #pandas #dataframe #numpy

#python #pandas #фрейм данных #numpy

Вопрос:

У меня есть серия pandas, которую я хочу преобразовать в массив numpy. Отображаются названия продуктов, но я не могу получить к ним доступ. Реальный вопрос заключается в поиске имен и, по сути, их сохранении в парах ключ-значение, но в массиве, а не в dict (формат, необходимый для круговой диаграммы)

 data = pd.read_csv('./data/clean_data/cleaned_data.csv')
ser = data.groupby(['Product']).sum()['QuantityOrdered']

 

Вывод

 Product
batteries                7661
charging cable          15187
...                       ...
eraser                   7538
 
 vals = ser.to_numpy()
 

Как это выглядит сейчас

 [7661, 15187, ..., 7538]
 

Как я хочу, чтобы это выглядело:

 [[batteries, 7661],[charging cable, 15187, ...]
 

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

1. Какую форму массива и dtype вы хотите?

Ответ №1:

Вы хотите ser.iteritems() (или list(ser.iteritems()) если вам напрямую нужны результаты в виде списка), а не ser.to_numpy() .