NUMPY064 题解：前 k 个元素求和

题目

计算数组前 $k$ 个元素的和。

解题思路

1. 读取数组和 $k$
2. 使用切片 $\texttt{Z[:k]}$ 获取前 $k$ 个元素
3. 求和

代码

import numpy as np

n = int(input())
arr = np.array(list(map(int, input().split())))
k = int(input())
result = arr[:k].sum()
print(result)

代码详解

• $\texttt{arr[:k]}$ —— 切片，获取前 $k$ 个元素
• $\texttt{.sum()}$ —— 求和