我有一个命名数组，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
我不知道这是否是一个错字，假设您的数组值看起来像字符串，假设不是，并且它们实际上是int，那么您可以通过调用来过滤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
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句