列表理解与可迭代。蟒蛇新手

Question

假设我有一个数组：

array=[1,0,2,3,4,0,0,5,6,0]

我想要一个只返回数字而不是零的列表。所以我做了这个并且它有效：

print(list(y for y in array if y!=0)

我尝试了另一种没有列表理解的方法，但它不起作用，有人可以解释为什么吗？

for y in array:
    if y!=0:
        print(list(y))

还有什么方法可以打印不带 0 的数字列表？

编辑：我尝试使用 for 循环来解决这个问题，如果我在顶部创建一个空列表，它就可以工作。它有效，但我不明白为什么！但是为什么这个有效而另一个无效呢？就速度/内存而言，这个或列表理解哪个更有效？

array=[1,0,2,3,4,0,0,5,6,0]
list=[]
for y in array:
    if y!=0:
        list.append(y)
print(list)

Answer 1

让我们好好看看你认为正确的地方：

for y in array:
    if y!=0:
        print(list(y))

所以我们遍历数组中的每个值。如果值不为零，我们打印list(y)。问题从这里开始。因为y 是整数，所以list(y) 会返回错误，因为您无法将整数转换为列表。如果您使用print(y)，它会起作用。

但随之而来的是另一个问题。如果我们打印列表中不是零的每个元素，我们会得到这样的结果，因为该代码只会按顺序打印：

1
2
3
4
5
6

您在问题中声明您想要一个列表。所以这段代码也不起作用，因为没有存储列表。所以我们终于得到了正确的答案：

array=[1,0,2,3,4,0,0,5,6,0]
list=[]
for y in array:
    if y!=0:
        list.append(y)
print(list)

此答案将每个不为零的y 值存储在列表中，然后最终打印出列表。

编辑：

这就是列表推导的工作原理：

首先，我不禁注意到您犯了语法错误。 :P 你忘了结尾括号！这是正确的代码： print(list(y for y in array if y != 0)).

其次，我需要声明那是不是列表理解。那是一个发电机。有细微的差别。

列表推导会在现场生成列表。列表理解如下所示：[y for y in array if y != 0]

生成器，也就是您在上面使用的，是存储的表达式。它看起来像这样：y for y in array if y != 0。

所以不用list(y for y in array if y != 0，你可以直接去[y for y in array if y != 0]。

所以现在我将解释“列表理解”（生成器表达式）的实际工作原理。它首先循环遍历array 的每个值。它检查该值是否不为零。如果不是，它会将y 添加到输出列表中。所以基本上，生成器表达式与带有 for 循环的第二个工作代码相同，除了 python 为您创建输出列表，增加了一些便利。

Answer 2

如果您想使用for 循环来执行此操作，您需要执行以下操作：

nonzero = []
for y in array:
   if y != 0:
      nonzero.append(y)
print(nonzero)

但是列表理解或filter 在可读性和速度方面都会更好。

Answer 3

在您的第二次尝试中，您拥有 list(y)，而您应该拥有 y。由于 y 是单个 int 而不是可迭代的，list(y) 失败。这是输出：

>>> array=[1,0,2,3,4,0,0,5,6,0]
>>> print(list(y for y in array if y!=0))
[1, 2, 3, 4, 5, 6]
>>> for y in array:
    if y!=0:
        print(list(y))


Traceback (most recent call last):
  File "<pyshell#49>", line 3, in <module>
    print(list(y))
TypeError: 'int' object is not iterable
>>> for y in array:
    if y!=0:
        print(y)


1
2
3
4
5
6

另一种方法是使用filter，像这样：

>>> print(list(filter(bool,array)))
[1, 2, 3, 4, 5, 6]

filter 接受一个函数和可迭代对象，将函数应用于可迭代对象中的每个项目，并仅返回函数返回真值的项目。如果 y 为 0（假），bool 为 False，否则为 True，它非常适合这些目的。不过，看看您也可以这样做可能对您有用：

>>> print(list(filter(lambda x:x != 0, array)))
[1, 2, 3, 4, 5, 6]

---

我知道你也想知道为什么第一个版本有效。

>>> print(list(y for y in array if y!=0))

这里，for y in array 是一个 generator，一种特殊的 Python 对象，每次你请求它时都会吐出一个值，可以这么说。所以让我这样分解：

print(                              )    #Print
      list(                        )     #Convert iterable to list
             for y in array              #For each y in array...
                            if y!=0      #If y is not zero...
           y                             #Include y in the iterable for list()

这有帮助吗？

Answer 4

当您创建for 循环时，该循环中的所有内容（所有内容）都会在每次循环中执行一次。所以这一行：

print(list(y)) 

...每次循环都打印一个新的list。而在第一个版本中：

print(list(y for y in array if y!=0)) 

...只打印了一件事：list() 理解的结果。

编辑：如果你想修复第二个版本中的'int' object is not iterable错误，你可以这样做：

for y in array:
    if y!=0:
        print([y])

list 函数需要一个可迭代对象作为输入。一个整数是不可迭代的，所以我们用括号括起来把它放在list 中，这基本上做同样的事情。使用上面的版本，您会看到每次循环都会创建一个新的list。

Answer 5

list() 是一个构造函数，它接受零个参数或一个可迭代对象，例如列表、str 或元组。问题是 int 不是可迭代的，其中作为列表推导式 ([y for y in array if y!= 0]) 返回可迭代并可用于创建列表Python documentation。