from collections import UserList, Iterable
class MyList(UserList):
def __getitem__(self, item):
if isinstance(item, Iterable):
return MyList(x for x in self.data if x in item)
return super(MyList, self).__getitem__(item)
if __name__ == '__main__':
l = MyList([1, 2, 3, 4, 5])
v = l[[2, 3, 11]]
print(v) # [2, 3]
通过定义customgetitem()方法,可以创建一个MyList类型,该类型将具有所需的行为
这些问题已在此处解决: Filter dataframe rows if value in column is in a set list of values
下面是一个使用.isin()方法的示例,相当于R的%in%。你知道吗
您是正确的:默认情况下,Python操作不是矢量化的。在这方面,R比常规Python更接近于第三方Pandas的API。因此可以使用熊猫系列对象:
Pandas建立在NumPy的基础上,因此毫不奇怪,您可以在NumPy中执行相同的操作:
相关问题 更多 >
编程相关推荐