我需要制作一个n×n的2D数组,如下所示
让我们以n等于5为例
n = 5
A = np.array([[1,2,3,4,5],[2,3,4,5,6],[3,4,5,6,7], [4,5,6,7,8], [5,6,7,8,9]])
我不知道该怎么做
您可以先创建一维序列数组,然后对其自身进行外加:
a = np.arange(5)
np.add.outer(a, a) + 1
#[[1 2 3 4 5]
# [2 3 4 5 6]
# [3 4 5 6 7]
# [4 5 6 7 8]
# [5 6 7 8 9]]
或a + a[:, None] + 1
给出相同的结果。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句