首页 > 解决方案 > 创建 numpy 数组问题(无法将输入数组从形状(2)广播到形状(1))

问题描述

创建这个numpy数组有什么问题

np.array( [np.array([1]), np.array([ [1,2] ])] )

# Error: could not broadcast input array from shape (2) into shape (1)

但这没问题

np.array( [np.array([1]), np.array([ [1,2], [1,2] ])] )

为不同形状的不同事物创建容器数组有什么问题?

标签: pythonarraysnumpy

解决方案


如果最终输出可以是一维向量,np.append则可以解决问题:

np.append(np.array([1]),np.array([[1,2]]))

如果最终所需数组的每个元素都是不同的维度,你需要它是一个 numpy 对象吗?清单应该有效final = [np.array([1]), np.array([ [1,2], [1,2] ])]


推荐阅读