重邮python实验课之创建ndarray

最新推荐文章于 2023-03-09 21:09:08 发布

Juhani_chen

最新推荐文章于 2023-03-09 21:09:08 发布

阅读量2.1k

点赞数 5

文章标签： python 开发语言

本文链接：https://blog.csdn.net/m0_62881629/article/details/124556965

版权

这段代码展示了如何利用NumPy库在Python中创建一个全为0的一维数组，并将指定位置的元素设为1。首先，通过`input().split()`读取用户输入的n和m，然后使用`np.zeros()`生成长度为n的0数组，接着将第m个元素（Python索引从0开始，所以是m-1）置为1。最后，将结果数组转换为二维数组输出。

摘要由CSDN通过智能技术生成

【问题描述】

创建一个长度为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()