NUMPY006 题解：创建特定位置为1的向量

题目

创建一个大小为 10 的向量，其中第 5 个元素（索引为 4）的值为 1，其余为 0。

解题思路

1. 先创建一个全 0 向量
2. 然后通过索引赋值，将索引为 4 的位置设为 1

代码

import numpy as np
Z = np.zeros(10)
Z[4] = 1
print(Z)

代码详解

1. np.zeros(10) —— 创建 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
2. Z[4] = 1 —— 将索引为 4 的位置设为 1
   • 注意：索引从 0 开始，所以第 5 个元素对应索引 4
3. 输出结果：[0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]

索引的基础知识

索引:  0   1   2   3   4   5   6   7   8   9
值:   [0.  0.  0.  0.  0.  0.  0.  0.  0.  0.]
       ↑
     第1个元素

核心知识点

1. 索引从 0 开始
2. 使用 array[index] = value 进行赋值
3. np.zeros() 创建全 0 数组

NumPy006 题解

题目分析

本题要求给定长度 n 和位置 k，创建长度为 n 的零向量，将第 k 位设为 1，然后打印数组和 shape。

解题思路

1. 从输入读取两个整数 n 和 k
2. 使用 np.zeros(n) 创建长度为 n 的零向量
3. 使用索引 arr[k] = 1 修改第 k 位
4. 打印数组和 shape

参考代码

import numpy as np

n, k = map(int, input().split())
arr = np.zeros(n)
arr[k] = 1
print(arr)
print(arr.shape)

输出结果

对于输入 10 4：

[0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]
(10,)