微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python – 运算符中的Numpy / pandas元素

我有一个列(或更确切地说是pandas帧)有一个列A,这列中的值是整数(假设它们属于范围1..10).

现在我必须在这个数组中选择A值为{3,6,9}的行(在这个例子中,它可以只是或==操作,但在现实生活中这个集合要长得多.

在任何一个库(pandas或numpy)中都有任何函数允许我快速执行以下操作:

arr = pandas.DataFrame(...)
values = [3, 6, 9] 
valid_indexes = magic_function(arr.A, values)

或者在numpy中:

arr = np.ndarray(...)
values = [3, 6, 9] 
valid_indexes = magic_function(arr[13, :], values)

换句话说,我正在寻找运算符中的元素.

解决方法:

docs是here

arr.loc[arr.A.isin([3,6,9])]

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐