python - 从 Pandas 数据框中获取数组的子集

Question

我有一个名为 arr 的 numpy 数组，其中包含 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

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

例如:

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

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

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