我可以使用 List Comprehensions 在 python 中的二维数组中获取输入吗？？以及如何为 nD 阵列实现它？

问题描述

假设用户想要输入以下格式。

如何在二维数组中获取这个输入？ nD（n = 自然数）数组的列表推导式的一般规则是什么？ 请推荐，谢谢。

解决方法

n = 4    
a = [[int(input()) for j in range(n)] for i in range(n)]

,

你可以这样试试：

out = []
for i in range(n):
    temp = list(map(int,input().split()))
    #temp will be [1,1,0] in the first iteration
    out.append(temp)

出站方式如下：

[[1,0],[0,[1,2,4,0]

,

为此您需要一个循环。

user_input = []
while True:
    row = input('Next row:\n')
    if len(row) == 0:
        break
    user_input.append([int(x) for x in row.split()])

,

可以，但必须提前知道维数。
喜欢 2D 4x4：

a=[[int(input()) for x in range(4)] for y in range(4)]

4-s 可以是任何东西，但嵌套本身在这样的代码中是固定的。

对于任意数量的维度，您可以尝试一些递归：

def nd(dims):
  if len(dims) == 0:
    return int(input())
  return [nd(dims[:-1]) for x in range(dims[-1])]

然后像运行一样

>>> nd([2,2])
1
2
3
4
[[1,2],[3,4]]

用于 2x2 2D
或

>>> nd([2,3,4])
1
2
3
[...]
23
24
[[[1,4],[5,6]],[[7,8],[9,10],[11,12]],[[13,14],[15,16],[17,18]],[[19,20],[21,22],[23,24]]]

用于 2x3x4 3D。