Numpy 一維數組到二維數組，第一個元素被隔離

Question

我從類似的東西開始

x = np.array([5, 1, 2, 3, 4], [6, 2, 4, 6, 8])

我想要的是這樣的：

x = np.array([5, [[1, 2], [3, 4]], [6, [[2, 4],[6, 8]] ]

我現在有東西

y = np.array([x[0][0], x[0][1:]], [x[1][0], x[1][1:]])

Answer 1

你錯過了[]

x = np.array([[5, 1, 2, 3, 4], [6, 2, 4, 6, 8]])

使用列表理解

[[row[0],row[1:].reshape(2,2).tolist()] for row in x]