我有一个列(或更确切地说是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] 举报,一经查实,本站将立刻删除。