我有一本字典,我想获得其值为最小非零值的密钥。
例如。鉴于输入:
{1:0, 2:1, 3:2}
它将返回2。
你可以在一次迭代中完成。
d = {1:0, 2:1, 3:2}
# Save the minimum value and the key that belongs to it as we go
min_val = None
result = None
for k, v in d.items():
if v and (min_val is None or v < min_val):
min_val = v
result = k
print(result)
一些假设:
min_val
将保持最小值您可以使用0
被认为是False
来过滤掉0
值。然后使用带有生成器表达式的next
:
d = {1:0, 2:1, 3:2}
val = min(filter(None, d.values()))
res = next(k for k, v in d.items() if v == val) # 2
这只会在1
为重复键的情况下返回一个键。对于多个匹配,您可以使用列表推导:
res = [k for k, v in d.items() if v == val]
请注意,您的文字要求“最小非零”将包含负值。
表现说明
上述解决方案是2遍,但具有时间复杂度O(n),不可能具有比这更低的复杂度。如@Maor所示,可以使用1遍O(n)解决方案,但这不一定更有效:
# Python 3.6.0
%timeit jpp(d) # 43.9 ms per loop
%timeit mao(d) # 98.8 ms per loop
%timeit jon(d) # 183 ms per loop
%timeit reu(d) # 303 ms per loop
用于基准测试的代码:
from random import randint
n = 10**6
d = {i: randint(0, 9) for i in range(n)}
def jpp(d):
val = min(filter(None, d.values()))
return next(k for k, v in d.items() if v == val)
def mao(d):
min_val = None
result = None
for k, v in d.items():
if v and (min_val is None or v < min_val):
min_val = v
result = k
return result
def jon(d):
return min({i for i in d if d[i] != 0})
def reu(d):
no_zeros = {k: v for k, v in d.items() if v != 0}
key, val = min(no_zeros.items(), key=itemgetter(1))
return key
假设dict
被命名为a
:
from operator import itemgetter
a = {1:0, 2:1, 3:2}
# remove zeros
no_zeros = {k: v for k, v in a.items() if v != 0} # can use `if v`
# find minimal key and value (by value)
key, val = min(no_zeros.items(), key=itemgetter(1))
# key = 2, val = 1
print(min(i for i in dictionary if dictionary[i] != 0))
这使得一个没有零的集合并返回该集合中的最小值。虽然值得指出这会导致2次迭代,因此比Maor Refaeli的解决方案慢。
解
some_dict = {1:0, 2:1, 3:2}
compare = []
for k, v in some_dict.items():
if k != 0:
compare.append(k)
x = min(compare)
print(x)
我只是appended
所有非零keys
到list
(compare
),然后应用min(compare)
我们可以插入x
并检查它是否指向关键1
这是最小的非零key
并返回它的value
这是0
>>> print(some_dict[x])
>>> 0