CQUPT Python numpy矩阵形状

【问题描述】

给定长度为12的列表a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]，输入参数n和m，如果能改变其形状为n*m，则输出改变后的数组，否则输出NO。

【输入形式】

整数n和m

【输出形式】

若满足n*m=12，输出数组，否则输出"NO"。

【样例输入】
4 3

【样例输出】

[[1 2 3]

 [4 5 6]

 [7 8 9]

 [10 11 12]]

【完整代码】

import numpy as np

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
a = np.array(a)
n, m = map(int, input().split())
if n * m == 12:
    print(a.reshape([n, m]))
else:
    print("NO")

【代码讲解】