NUMPY068 题解：分割数组

题目

将数组分割成 $k$ 个相等的部分。

解题思路

1. 读取数组和分割数 $k$
2. 使用 $\texttt{np.split}$ 进行分割

代码

import numpy as np

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

代码详解

$\texttt{np.split(arr, k)}$ —— 将数组 $arr$ 分割成 $k$ 个近似相等的部分。

运行示例

输入：

6
1 2 3 4 5 6
2

分析：

• 将数组 $[1, 2, 3, 4, 5, 6]$ 分成 $2$ 部分
• 每部分约 $6 \div 2 = 3$ 个元素

输出：

[[1 2 3]
 [4 5 6]]

核心知识点

1. $\texttt{np.split}$ —— 分割数组
2. 分割点位置要均匀分布

NUMPY068 题解：分割数组

题目

将数组分割成 $k$ 个相等的部分。

解题思路

1. 读取数组和分割数 $k$
2. 使用 $\texttt{np.split}$ 进行分割

代码

import numpy as np

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

代码详解

$\texttt{np.split(arr, k)}$ —— 将数组 $arr$ 分割成 $k$ 个近似相等的部分。