如何在Python中如下创建嵌套列表？

Question

我想创建如下所述的嵌套列表：

a MOD b等于零，然后在该索引中填充1，否则填充零

MOD运算符为％，因此您需要执行A％B，还请注意0％0或0％1会引发异常，因此请找到此提示的解决方案添加+ 1

预期输出：

[[1,0,0],
 [0,1,0],
 [0,0,1]]

但是，我知道如何使用3x3创建嵌套列表，如下所示，但不知道下一步如何做：

a = 3, b = 3
print([[0 for n in range(a)] for m in range(b)])

谢谢您的帮助！

Answer 1

import numPy as np
b = np.identity(3, dtype = float) 
print(b)