【问题描述】给定长度为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 def main(): a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] n, m = input().split() n = int(n) m = int(m) a = np.array(a) 改变形状代码 if __name__ == '__main__': main()

【样例程序片段解析】

该程序片段利用numpy库将列表a转换成了numpy数组，并利用numpy数组的reshape方法改变了其形状。具体解释见代码注释。

【样例程序片段实现】

import numpy as np

def main():
    a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
    n, m = input().split()
    n = int(n)
    m = int(m)
    a = np.array(a)
    # 判断能否改变形状
    if n * m != 12:
        print("NO")
    else:
        # 改变形状为n*m
        a = a.reshape(n, m)
        print(a)

if __name__ == '__main__':
    main()