Python 字典初始化和值附加到列表



问题:编写函数,对于给定的对象列表,返回类型为:[对象]的字典。 例如:func([1,2.0,3,'g']) returns: {'i':[1,3], 'f': [2.0], 'c': ['g'], 'o': []}

  • 整数的关键i
  • 浮子的关键f
  • 字符的关键c
  • 对他人的关键o

我的代码:

def make_dict(lst):
my_dict = {'i': [], 'f': [], 'c': [], 'o': []}
[my_dict['i'].append(item) if isinstance(item, int) else my_dict['f'].append(item) if isinstance(
item, float) else my_dict['c'].append(item) if isinstance(item, str) and len(item) == 1 else my_dict['o'].append(item) for item in lst]
return my_dict

这个函数返回相应的字典,但我需要把它变成一个单行,另外,我不喜欢它的方式。我知道在这里使用列表理解是不好的。

我的过程是首先使它成为一个多行函数,我使用 for 循环和 if elif 语句,然后我尝试组合这些行并最终将其减少到 3。

这是我在将其组合成列表理解之前使用的 for 循环:

for item in lst:
if isinstance(item, int):
my_dict['i'].append(item)
elif isinstance(item, float):
my_dict['f'].append(item)
elif isinstance(item, str) and len(item) == 1:
my_dict['c'].append(item)
else:
my_dict['o'].append(item)
return my_dict

我的问题: 我知道我需要以某种方式使用某种字典理解与 filter(( 的 map(( 相结合,但在搜索互联网并找到注释之后。我只是不知道该怎么做。

也许你可以帮我?谢谢!

您想要的是按其类型的第一个字母对数据进行分组,以便

  • 获取其类型的第一个字母type(x).__name__[0]
  • 按此条件对数据进行分组

您需要提供数据,以便按用于分组的相同条件进行排序groupby{k:list(v)部分是从groupby对象移动到dict

from itertools import groupby
def make_dict(lst):
sorter = lambda x:type(x).__name__[0] if type(x).__name__ in ('int', 'float', 'str') else 'o'
return {k:list(v) for k,v in groupby(sorted(lst, key=sorter), key=sorter)}
r = make_dict([1,2.0,3,'g']) # {'f': [2.0], 'i': [1, 3], 's': ['g']}
print(r)

使用filter,但效率不高,因为多次迭代列表

def make_dict(lst):
return {'i': list(filter(lambda x: isinstance(item, int), lst)), 
'f': list(filter(lambda x: isinstance(item, float), lst)), 
's': list(filter(lambda x: isinstance(item, str), lst)), 
'o': type(x).__name__ not in ('int', 'float', 'str')}

最新更新