问题:编写函数,对于给定的对象列表,返回类型为:[对象]的字典。 例如: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')}