[英]Python: List comprehension not working correctly
以下是学习过程的一部分,非常感谢您的帮助!
我在对列表理解进行反向工程时遇到问题。 我有一个输入数据列表:
tableData = [['apples', 'oranges', 'cherries', 'banana'],
['Alice', 'Bob', 'Carol', 'David'],
['dogs', 'cats', 'moose', 'goose']]
我想创建一系列新的列表,如下所示:
['apples', 'Alice', 'dogs']
['oranges', 'Bob', 'cats']
['cherries', 'Carol', 'moose']
['banana', 'David', 'goose']
我可以使用:
i = 0
for li in range(4):
out = [item[i] for item in tableData]
print(out)
i += 1
但是,当我尝试使用时:
i = 0
out = []
for li in range(4):
for item in tableData:
out.append(item[i])
print(out)
i += 1
这会导致错误。
知道为什么吗? 如何使它像以前的示例一样工作?
错误背后的原因已由@Matt B引用。
但是,这里的有效方法可能是:
print(list(zip_longest(tableData[0],tableData[1], tableData[2])))
甚至更好,是广义的。 (感谢@Patrick Haugh)
print(list(zip_longest(*tableData)))
输出 :
[('apples', 'Alice', 'dogs'), ('oranges', 'Bob', 'cats'), ('cherries', 'Carol', 'moose'), ('banana', 'David', 'goose')]
请注意,我使用
zip_longest
而不是zip
来处理额外的元素,zip_longest
,例如(注意red apples
和Elon
):
tableData = [['red apples', 'apples', 'oranges', 'cherries', 'banana'],
['Alice', 'Bob', 'Carol', 'David', 'Elon'],
['dogs', 'cats', 'moose', 'goose']]
使用zip
会错过banana
和Elon
的输出:
[('red apples', 'Alice', 'dogs'), ('apples', 'Bob', 'cats'), ('oranges', 'Carol', 'moose'), ('cherries', 'David', 'goose')]
但是使用longest_zip
会将缺少的值插入为None
:
[('red apples', 'Alice', 'dogs'), ('apples', 'Bob', 'cats'), ('oranges', 'Carol', 'moose'), ('cherries', 'David', 'goose'), ('banana', 'Elon', None)]
引发错误是因为您进行了4次循环( range(4)
),以循环内部只有3个元素的tableData
。
i = 0
out = []
for li in range(3): # <<< Change to 3
for item in tableData:
out.append(item[i])
print(out)
i += 1
更好的方法是在tableData
的len
range
循环,这样它可以动态运行,并且可以更改tableData
的长度,而不会抛出我认为是IndexError
:
i = 0
out = []
for li in range(len(tableData))
for item in tableData:
out.append(item[i])
print(out)
i += 1
如果要打印出tableData
内的列表,则可以编写:
for item in tableData:
print(item)
Matt B的示例:
i = 0
out = []
for li in range(len(tableData)):
for item in tableData:
out.append(item[i])
print(out)
i += 1
这也会导致错误,因为i
会越界。 您不必初始化i
,只需添加item
。
out = []
# No nested for loop. this will cause duplicates of the list
for item in tableData:
out.append(item) #<<< No need for item[i]
print(out)
如果您已经有一个列表列表,则无需创建新的列表列表。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.