根据值拆分Numpy数组 [英] Splitting Numpy array based on value

问题描述

假设我有以下NumPy数组:

Suppose I have this NumPy array:

a = np.array([0, 3, 5, 5, 0, 10, 14, 15, 56, 0, 12, 23, 45, 23, 12, 45, 
              0, 1, 0, 2, 3, 4, 0, 0 ,0])

我想打印0之间的所有数字，并将它们自动添加到新的np.array(见下文):

I would like to print all the numbers between 0s and automatically add them to a new np.array (see below):

a1=[3, 5, 5]
a2=[10, 14, 15, 56]
a3=[12, 23, 45, 23, 12, 45]
a4=[1]
a5=[2, 3, 4]

是否有内置功能可以做到这一点?

Is there a built-in function to do this?

推荐答案

您可以使用itertools中的groupby()函数，并将key指定为零或非零的布尔条件.这样，所有连续的零和非零将被组合在一起.使用if过滤器拾取非零的组，并使用list将非零的分组者转换为列表.

You can use groupby() function from itertools, and specify the key as the boolean condition of zero or nonzero. In such a way, all consecutive zeros and nonzeros will be grouped together. Use if filter to pick up groups of nonzeros and use list to convert the non zero groupers to lists.

from itertools import groupby
[list(g) for k, g in groupby(a, lambda x: x != 0) if k]

# [[3, 5], [10, 14, 15, 56], [12, 23, 45, 23, 12, 45], [1], [2, 3, 4]]

这篇关于根据值拆分Numpy数组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！