从 pandas 数据框中获取数组的子集

Question

我有一个 numpy 数组，arr其中包含 1154 个元素。

array([502, 502, 503, ..., 853, 853, 853], dtype=int64)

我有一个名为的数据框df

    team    Count
0   512     11
1   513     21
2   515     18
3   516     8
4   517     4

如何获取df仅包含数组中的值的数据框的子集arr

例如：

team         count
arr1_value1    45
arr1_value2    67

为了使这个问题更清楚：我有一个 numpy 数组['45', '55', '65']

我有一个数据框如下：

team  count
34      156
45      189
53       90
65       99
23       77
55       91

我需要一个新的数据框，如下所示：

team    count
 45      189
 55       91
 65       99

Answer 1

我不知道这是否是一个拼写错误，你的数组值看起来像字符串，假设它不是并且它们实际上是整数，那么你可以通过调用来过滤你的 df isin：

In [6]:

a = np.array([45, 55, 65])
df[df.team.isin(a)]
Out[6]:
   team  count
1    45    189
3    65     99
5    55     91