Question

我有一个如下所示的列表：

[[＆＃39; 1＆＃39;，＆＃39; H＆＃39;，＆＃39;氢气＆＃39;，＆＃39; 1.00794（4）＆＃39;，＆＃39; FFFFFF＆＃39;，＆＃39; 1s1＆＃39;，＆＃39; 2.2＆＃39;，＆＃39; 37＆＃39;，＆＃39 ;＆＃39;，＆＃39; 120＆＃39;，＆＃39; 1312＆＃39;，＆＃39; -73＆＃39;，＆＃39; -1,1＆＃39;，＆＃39; gas＆＃39;，＆＃39; diatomic＆＃39;＆＃39; 14＆＃39;＆＃39; 20＆＃39;，＆＃39; 0.0000899＆＃39;，＆＃39; nonmetal＆＃39; ，＆＃39; 1766＆＃39;]，[＆＃39; 2＆＃39;，＆＃39;他＆＃39;，＆＃39;氦气＆＃39;，＆＃39; 4.002602（2）＆＃39;，＆＃39; D9FFFF＆＃39;，＆＃39; 1s2＆＃39;，＆＃39;＆＃39;，＆＃39; 32＆＃39;，＆＃39; ＆＃39;，＆＃39; 140＆＃39;，＆＃39; 2372＆＃39;，＆＃39; 0＆＃39;，＆＃39;＆＃39;，＆＃39; gas＆＃39;，＆＃39; atomic＆＃39;，＆＃39;＆＃39;，＆＃39; 4＆＃39;，＆＃39; 0.0001785＆＃39;，＆＃39;贵气“＆＃39;，＆＃39; ; 1868＆＃39;]等......]

一直沿着元素周期表走。我还有一个函数（在一个类中，因此是self。），它应该打印一个子列表（每个元素一个），如下所示：

def get_element_data(self, identifier):
    if type(identifier) == int:
        if self.check_atomic_number_exists(identifier):
            return self.parsed_data[identifier - 1]
    if type(identifier) == str:
        if len(identifier) <= 2:
            print("passed in symbol")
            return (element for element in self.parsed_data if identifier == 
                   element[1])
        if len(identifier) > 2:
            print("passed in name")
            return (element for element in self.parsed_data if identifier == 
                    element[2])

parsing_object.get_element_data("Hydrogen")

但如果我运行它，则返回生成器对象而不是子列表。如何让它返回子列表？此外，如果我按原子序数搜索它也能正常工作。

Answer 1

让它返回一个列表。

return list(element for ...)

...

return [element for ...]

如果那就是你所需要的，甚至是一个元组。

return tuple(element for ...)

Answer 2

你要归还这个： (element for element in self.parsed_data if identifier == element[2]) 这实际上是一个发电机。我不太了解短python，所以我不确定我是否完全理解你的代码，但我认为这就是你想要做的事情：

foreach(element in elements):
     if identifier == element[2]:
        return element

如果您希望它的行为类似于生成器，则不应使用return，您应该使用yield。这使你能够做到：

foreach(value in generator()):
  print value

所以对你来说就是

foreach(element in elements):
     if identifier == element[2]:
        yield element

使用给定列表，如何使此搜索函数返回值而不是它的生成器对象

2 个答案: