如果列表不是None,则获取列表的第一个元素:Python

时间:2012-02-17 11:11:09

标签: python list

这是我的问题:

我在Python中进行LDAP搜索。搜索将返回一个字典对象:

{'mail':['user@mail.com'],'mobile':['07852242242'], 'telephoneNumber':['01112512152']}

如您所见,返回的字典包含列表值。

有时候,找不到任何结果:

{'mail':None, 'mobile':None, 'telephoneNumber':['01112512152']}

要提取所需的值,我使用get()以避免在字典项不存在时出现异常。

return {"uname":x.get('mail')[0], "telephone":x.get('telephoneNumber')[0], "mobile":x.get('mobile')[0]}

我想返回我自己的字典,只有字符串值,但我很难找到一种有效的方法来检查列表是否为None并继续遇到索引错误或输入错误:

(<type 'exceptions.TypeError'>, TypeError("'NoneType' object is unsubscriptable",)

有没有办法在列表中使用get()方法,这样如果列表为None,它就不会抛出异常???

{"uname":x.get('mail').get(0)}

获取列表的第一个值或返回None的最有效方法是什么,而不使用:

if isinstance(x.get('mail'),list):

if x.get('mail') is not None:

4 个答案:

答案 0 :(得分:3)

如果你想弄平字典,你可以这样做:

>>> d = {'mail':None, 'mobile':None, 'telephoneNumber':['01112512152']}
>>> 
>>> dict((k,v and v[0] or v) for k,v in d.items())
{'mail': None, 'mobile': None, 'telephoneNumber': '01112512152'}

如果您还希望过滤,切断None值,那么您可以这样做:

>>> dict((k,v[0]) for k,v in d.items() if v)
{'telephoneNumber': '01112512152'}

答案 1 :(得分:2)

尝试下一个:

return {"uname":(x.get('mail') or [None])[0], ...

它有点不可读,所以你可能想把它包装成一些辅助函数。

答案 2 :(得分:0)

你可以这样做:

input_dict = {'mail':None, 'mobile':None, 'telephoneNumber':['01112512152']}

input_key_map = {
    'mail': 'uname',
    'telephoneNumber': 'telephone',
    'mobile': 'mobile',
}

dict((new_name, input_dict[old_name][0]) 
      for old_name, new_name in input_key_map.items() if input_dict.get(old_name))

# would print:
{'telephone': '01112512152'}

答案 3 :(得分:0)

我不确定是否有直接的方式来做到这一点,但你可以试试:

 default_value = [None]
 new_x = dict((k, default_value if not v  else v) for k, v in x.iteritems())

并使用new_x代替x