我尝试使用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数组,使每个子数组的长度等于条件值?
1条答案
按热度按时间r6hnlfcb1#
使用
np.reshape
以所需的方式更改数组拆分对于
conditional_value
的更一般情况