根据默认大小重塑np.array

问题描述 投票:2回答:1

有2个np.arrays,我想重塑np.array1从形状(12,)in引用array2形状(4,)

array1 = np.array([ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])array1.shape回归:(12,)

array2 = np.array([ 12, 34, 56, 78])array2.shape回归:(4,)

我试图执行reshape

array1.reshape(array2.shape)

但是,有一个错误:ValueError: cannot reshape array of size 12 into shape (4,)

因此,预期的结果是array1有4个元素:np.array([ 1, 2, 3, 4]),而不是12。

我很感激任何想法和帮助。

python numpy reshape numpy-ndarray valueerror
1个回答
1
投票

如果我正确理解您的要求,我认为您正在寻找的是简单的切片:

In [140]: array2 = np.array([ 12,  34,  56,  78])
In [135]: a_sliced =  array1[:array2.shape[0]]
In [136]: a_sliced.shape
Out[136]: (4,)

如果array2是多维的,那么使用approach suggested by Mad Physicist

sliced_arr = array1[tuple(slice(0, d) for d in array2.shape)]

或者,如果您打算将数组拆分为三个相等的一半,则使用numpy.split(),如下所示:

# split `array1` into 3 portions
In [138]: np.split(array1, 3)
Out[138]: [array([1, 2, 3, 4]), array([5, 6, 7, 8]), array([ 9, 10, 11, 12])]
© www.soinside.com 2019 - 2024. All rights reserved.