根据numpy中的条件拆分数组,可以使用numpy的numpy.split()
函数。以下是一个示例:
import numpy as np
# 创建一个numpy数组
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
# 定义拆分条件
condition = arr % 2 == 0
# 使用numpy.split()函数拆分数组
splitted_arrays = np.split(arr, np.where(condition)[0])
# 输出拆分后的数组
print(splitted_arrays)
在这个示例中,我们首先创建了一个numpy数组,然后定义了一个拆分条件,即数组中的偶数元素。接下来,我们使用numpy.split()
函数将数组拆分为多个子数组,其中np.where(condition)[0]
用于找到满足条件的元素的索引。最后,我们输出拆分后的数组。
这种方法可以很容易地根据任何条件拆分numpy数组。只需要修改condition
变量即可。
领取专属 10元无门槛券
手把手带您无忧上云