CQUPT Python 创建ndarray

CQUPT_114514

于 2023-04-17 16:44:10 发布

阅读量434

点赞数 2

分类专栏： Python程序设计文章标签： python

本文链接：https://blog.csdn.net/2301_76172695/article/details/130203687

版权

27 篇文章 45 订阅

订阅专栏

该代码示例展示了如何在Python中利用numpy库创建一个长度为n的一维全零数组，并将第m个元素设置为1。首先，通过`input().split()`获取用户输入的n和m，然后使用`np.zeros`创建数组，并通过索引`arr[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

n, m = map(int, input().split())
arr = np.zeros([1, n], dtype=float)
arr[0][m - 1] = 1.
print(arr)

【代码讲解】

CQUPT_114514

关注

专栏目录