1

我想从列表中提取一个项目,如果找不到则返回默认值(而不是通常的索引异常)。与字典中的 get(key, default) 非常相似。仔细阅读文档后,感觉我仍然忽略了一个提供此功能的简单的内置 python 解决方案。

这是我能想到的最快的:

def pluck(list_items, index, default=None):
  return dict(zip(range(len(list_items)), list_items)).get(index, default)

s = [1, 2,]
pluck(s, 3, None)

谢谢!

4

4 回答 4

5

根据您希望尝试访问数组边界之外的频率,您应该捕获异常(如果预计失败很少见),或者测试索引是否有效,否则返回默认值。

如果预计故障很少见,这可能是最快的解决方案:

def pluck(items, index, defval):
    try:
        return items[index]
    except IndexError:
        return defval

如果预计它们会很频繁,最好检查索引是否有效:

def pluck(items, index, defval):
    return items[index] if -len(items) <= index < len(items) else defval

请注意,如果abs(index) <= len(items),即,使用负索引是有效的。items[-1]相当于items[len(items)-1]

于 2013-06-13T21:45:24.773 回答
3

如果您只使用正索引,为了保持快速和简单:

seq, i = [1, 2, 3], 2
plucked = seq[i] if seq[i:] else None

否则:

def pluck(seq, index, default=None):
    try:
        return seq[index]
    except IndexError:
        return default

如果例外是常态:

def pluck(seq, index, default=None):
    return seq[index] if seq[index:] and seq[-index-1:] else default

两者都适用于负索引。

于 2013-06-13T21:45:05.007 回答
2

像这样的东西,它也适用于负索引。

def list_get(lis, index, default = None):
   index = index if index >=0 else len(lis) + index
   return lis[index] if 0 <= index < len(lis) else default

>>> lis = range(10)
>>> list_get(lis, 8, 'foo')
8
>>> list_get(lis, 11, 'foo')
'foo'
>>> list_get(lis, -11, 'foo')
'foo'
>>> list_get(lis, -9, 'foo')
1
>>> list_get(lis, -5, 'foo')
5
于 2013-06-13T21:42:48.090 回答
1

看起来你的 pluck 函数可以写成

def pluck(list_items, index, default=None):
  if index>length(list_items)-1
     return default
   else 
      return list_items[index]
于 2013-06-13T21:44:34.927 回答