重邮python实验课之创建ndarray

【问题描述】

创建一个长度为n的一维全为0的ndarray对象，然后让第m个元素等于1 

输入n,m并用空格隔开, 且n>=m

【输入形式】
【输出形式】
【样例输入】

10 5
【样例输出】

[[0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]]

【样例说明】
【评分标准】

import numpy as np

def main():

    n, m = input().split(" ")
    a = np.zeros(eval(n))
    # b = np.array()
    a[eval(m)-1] = 1
    a = np.array([a])
    print(a)
if __name__ == "__main__":
    main()