基于条件值拆分分类numpy数组

x759pob2  于 2023-03-12  发布在  其他
关注(0)|答案(1)|浏览(140)

我尝试使用based基于条件值拆分一个分类numpy数组,这样拆分的子数组的长度就等于条件值。
例如,我有一个分类numpy数组,如下所示,我想根据条件值拆分numpy数组,如下所示

import numpy as np

x = np.array(['Horses', 'Cats', 'Dogs'])

conditional_value = 1

最终会得到一个numpy数组,如下所示:

Expected Output :

array([['Horses'],
       ['Cats'],
       ['Dogs']], dtype='<U6')

我尝试使用np.split,但结果是出乎意料的。

import numpy as np

x = np.array(['Horses', 'Cats', 'Dogs'])

np.split(x, 1)

Actual Output : [array(['Horses', 'Cats', 'Dogs'], dtype='<U6')]

还有什么方法可以拆分上面的numpy数组,使每个子数组的长度等于条件值?

r6hnlfcb

r6hnlfcb1#

使用np.reshape以所需的方式更改数组拆分

import numpy as np

x = np.array(['Horses', 'Cats', 'Dogs'])

np.reshape(x, [3,1])

对于conditional_value的更一般情况

import numpy as np

x = np.array(['Horses', 'Cats', 'Dogs','Owls'])
conditional_value = 2

np.reshape(x, [len(x)//conditional_value,conditional_value])

相关问题