✨Python小技巧 | 🎯 np.split()函数示例✨

在数据分析和机器学习中，`numpy` 是一个强大的工具库。今天来聊聊它的 `np.split()` 函数！🎉 `np.split()` 用于将数组沿指定轴分割成多个子数组，简单实用。👀

例如，我们有一个简单的数组 `[1, 2, 3, 4, 5, 6]`，想把它分成两部分： `[1, 2, 3]` 和 `[4, 5, 6]`。代码如下👇：

```python

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])

result = np.split(arr, 2)

print(result)

```

输出结果为：`[array([1, 2, 3]), array([4, 5, 6])]` 🎉

此外，如果想按具体位置分割，比如在索引 2 和 4 处分割，可以这样写：

```python

arr = np.array([1, 2, 3, 4, 5, 6])

result = np.split(arr, [2, 4])

print(result)

```

输出为：`[array([1, 2]), array([3, 4]), array([5, 6])]` 🌟

`np.split()` 灵活方便，适合多种场景！快试试吧！💪