用于具有变量递增和 2d 列表搜索的 2d 列表的一行 for 循环

Question

我是 python 单线循环的新手。

我希望用户将数据输入到二维列表中，同时提醒他他们将输入的数据索引。 我的代码是：

flag=0
x=[[int(input("enter the "+ str(flag)+ "number:")) flag+=1 for x in range(0,col)]for y in range(0,row)] 

以上代码显示语法错误。我应该如何在 for 循环的一行中增加标志值。

接下来我想在二维列表中搜索整数值

x=[[0,3,2],[1,1,1],[9,2,2]]
for i,j in enumerate(x):
    try:
        y=j.index(1)
    except ValueError:
        continue
    print(i,y)

上面的代码适用于单个项目搜索，但我想要列表中每个 1 的索引。而上面的代码只返回它遇到的第一个“1”的索引，即“1 0”

提前谢谢你。

Answer 1

您的问题是列表推导必须产生一个值，而您正试图在其中执行 flag +=1。

如果你这样做会更清楚：

>>> flag = 0
>>> row = 1
>>> col = 3
>>> x=[[int(input("enter the number [{}][{}]:".format(y, x))) for x in range(0,col)] for y in range(0,row)]
enter the number [0][0]:1
enter the number [0][1]:2
enter the number [0][2]:3
>>> x
[[1, 2, 3]]

或者，要完全实现您想要的，您可以执行以下操作：

>>> y=[[int(input("enter the {} number:".format(x+y*row))) for x in range(0,col)]for y in range(0,row)]
enter the 0 number:1
enter the 1 number:2
enter the 2 number:3
>>> y
[[1, 2, 3]]

问题第二部分的更新： 您可以为此使用更多列表推导：

>>> y=[[int(input("enter the {} number:".format(x+y*row))) for x in range(0,col)]for y in range(0,row)]
enter the 0 number:0
enter the 1 number:1
enter the 2 number:1
>>> y
[[0, 1, 1]]
>>> for arr in y:
...   print [index for (index, val) in enumerate(arr) if val == 1]
...
[1, 2]