您可以使用NumPy的split函数来按照索引的和将数组的元素分组。这里是一个示例代码:
import numpy as np
# 创建一个示例数组
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 定义索引的和用于分组
group_indices = [3, 5, 8]
# 使用split函数按照索引的和分组
groups = np.split(arr, group_indices)
# 打印分组结果
for group in groups:
print(group)
输出结果为:
[1 2 3]
[4 5]
[6 7 8]
[9 10]
在示例代码中,我们首先创建了一个示例数组arr。然后,我们定义了索引的和group_indices,它表示我们希望将数组元素分成4个组,分别是[1, 2, 3]、[4, 5]、[6, 7, 8]和[9, 10]。最后,我们使用split函数按照group_indices进行分组,并将分组结果打印出来。
上一篇:按照索引查找匹配括号的令牌
下一篇:按照索引的三个项目进行分组的分页