20

在 Python 中,是否可以完全封装常见的切片语法并传递它?我知道我可以使用slice__slice__模拟切片。但我想传递与 . 一起使用的方括号中完全相同的语法__getitem__

例如,假设我编写了一个函数来返回列表的某个片段。

def get_important_values(some_list, some_condition, slice):
    elems = filter(some_condition, some_list)
    return elems[slice]

如果我手动传入一个切片对象,这很好用:

In [233]: get_important_values([1,2,3,4], lambda x: (x%2) == 0, slice(0, None))
Out[233]: [2, 4]

但是我想让用户通过的切片与他们使用的切片完全相同__getitem__

get_important_values([1,2,3,4], lambda x: (x%2) == 0, (0:-1) )

# or

get_important_values([1,2,3,4], lambda x: (x%2) == 0, (0:) )

显然这会产生语法错误。但是有什么方法可以使这项工作,而不是为x:y:t类型切片编写我自己的迷你解析器,并强制用户将它们作为字符串传递?

动机

我可以让这个示例函数返回一些可直接切片的内容,例如filter(some_condition, some_list),这将是一个列表的整个结果。然而在我的实际例子中,内部函数要复杂得多,如果我提前知道用户想要的切片,我可以大大简化计算。但我希望用户不必做太多额外的事情来提前告诉我切片。

4

3 回答 3

15

也许以下内容对您有用:

class SliceMaker(object):
  def __getitem__(self, item):
    return item

make_slice = SliceMaker()

print make_slice[3]
print make_slice[0:]
print make_slice[:-1]
print make_slice[1:10:2,...]

这个想法是您使用make_slice[]而不是手动创建slice. 通过这样做,您将能够使用熟悉的方括号语法。

于 2012-12-04T15:37:15.700 回答
2

简而言之,没有。该语法仅在[]运算符的上下文中有效。我可能会建议接受一个元组作为输入,然后将该元组传递给slice(). 或者,也许您可​​以重新设计您正在做的任何事情,以便get_important_values()以某种方式实现为可切片对象。

例如,您可以执行以下操作:

class ImportantValueGetter(object):
    def __init__(self, some_list, some_condition):
        self.some_list = some_list
        self.some_condition = some_condition

    def __getitem__(self, key):
        # Here key could be an int or a slice; you can do some type checking if necessary
        return filter(self.some_condition, self.some_list)[key]

您可以通过将其变成某种容器 ABC 来做得更好,但这是一般的想法。

于 2012-12-04T15:34:21.430 回答
1

一种方法(对于简单切片)是让切片参数是 dict 或 int,

IE

get_important_values([1, 2, 3, 4], lambda x: (x%2) == 0, {0: -1})

或者

get_important_values([1, 2, 3, 4], lambda x: (x%2) == 0, 1)

那么语法将或多或少保持不变。

但是,这行不通,因为当您想做类似的事情时

some_list[0:6:10..]
于 2012-12-04T15:35:36.240 回答