NUMPY082 题解：重复数组元素

题目

将数组的每个元素重复 $k$ 次。

解题思路

1. 读取数组和重复次数
2. 使用 $\texttt{np.repeat}$ 重复元素

代码

import numpy as np

n, k = map(int, input().split())
arr = np.array(list(map(int, input().split())))
result = np.repeat(arr, k)
print(result)

代码详解

$\texttt{np.repeat(arr, k)}$ —— 将 $arr$ 中每个元素重复 $k$ 次。

运行示例

输入：

3 2
1 2 3

分析：

• $1$ 重复 $2$ 次：$[1, 1]$
• $2$ 重复 $2$ 次：$[2, 2]$
• $3$ 重复 $2$ 次：$[3, 3]$

输出：

[1 1 2 2 3 3]

核心知识点

1. $\texttt{np.repeat}$ —— 重复数组元素