好的,所以我必须创建这样的东西:
输入板的尺寸:8
1 0 0 0 0
0 1 0 0 0 0
0 0 1 0 0 0 0
0 0 0 1 0 0 0
0 0 0 0 1 0 0
0 0 0 0 1 0 0
0 0 0 0 1 0
0 0 0 0 1
我的问题是,我的代码只会创建一个0的矩形,我不知道如何添加1的对角线。
到目前为止我的代码:
dimension=int(input('Enter dimension of triangle: '))
if dimension < 2:
print('Invalid input')
else:
for r in range(dimension):
for c in range(dimension):
print("0",end=" ")
print()
对于对角线,行索引等于列索引,因此可以执行:
dimension=int(input('Enter dimension of triangle: '))
if dimension < 2:
print('Invalid input')
else:
for r in range(dimension):
for c in range(dimension):
# If we are on the diagonal
if r == c:
num = "1"
# Otherwise
else:
num = "0"
print(num, end=" ")
print()
>>> dimension = 8
>>> for r in range(dimension):
... for c in range(dimension):
... print(int(c==r), end=" ")
... print()
...
1 0 0 0 0 0 0 0
0 1 0 0 0 0 0 0
0 0 1 0 0 0 0 0
0 0 0 1 0 0 0 0
0 0 0 0 1 0 0 0
0 0 0 0 0 1 0 0
0 0 0 0 0 0 1 0
0 0 0 0 0 0 0 1
对于一般情况,您可以将备选方案放入print
函数中,如
print("1" if c==r else "0", end=" ")
a = np.zeros((8, 8), int)
np.fill_diagonal(a, 1)
如果你想打印
s = StringIO()
np.savetxt(s,a,fmt="%d")
print s.getvalue()