将系列转换为数组答案

【问题标题】：converting series to array将系列转换为数组
【发布时间】：2019-07-05 12:32:30
【问题描述】：

我只是在玩一个餐厅数据集，我使用count_values() 计算了不同餐厅位置的值，现在我需要将其提取为数组形式我的代码：

location_to_keep = dataset['location'].value_counts()

print(location_to_keep)

output:
BTM                           2181
Koramangala 5th Block         1987
Indiranagar                   1394
HSR                           1329
Jayanagar                     1281
JP Nagar                      1163
Whitefield                     916
Koramangala 7th Block          838
Koramangala 6th Block          813
Marathahalli                   762
Koramangala 4th Block          706
Brigade Road                   699
MG Road                        641
Bannerghatta Road              609
Ulsoor                         597
Koramangala 1st Block          568
Bellandur                      542
Sarjapur Road                  534
Kalyan Nagar                   532
Banashankari                   480
Residency Road                 465
Church Street                  464
Richmond Road                  457
Malleshwaram                   454
Lavelle Road                   437
Basavanagudi                   416
Electronic City                386
Cunningham Road                383
New BEL Road                   338
Frazer Town                    33

需要以数组形式提取餐厅名称吗？有些回复很快....

【问题讨论】：

欢迎来到 Stack Overflow。请阅读the help pages，获取the SO tour，了解how to ask good questions，以及this question checklist。最后请学习如何创建minimal reproducible example 向我们展示。
list(location_to_keep.index)?

标签： python pandas

【解决方案1】：

我认为@Quang Hoang 是正确的，list(location_to_keep.index) 应该可以解决您的问题。为了完整起见，这里是一个最小的可重现示例。

import pandas as pd
import numpy as np
dataset = pd.DataFrame()
locations = ['A', 'B', 'C', 'D']
dataset['location'] = np.random.choice(locations, 1000)

通过使用value_counts，我们应该得到如下所示的内容。

In [1]: location_to_keep = dataset['location'].value_counts()
        print(location_to_keep)
Out[1]: B    272
        C    259
        D    247
        A    222
        Name: location, dtype: int64

然后您可以使用index 获取名称，使用values 获取计数。

In [2]: list(location_to_keep.values)
Out[2]: [272, 259, 247, 222]

In [3]: list(location_to_keep.index)
Out[3]: ['B', 'C', 'D', 'A']

【讨论】：