Numpy 一维数组到二维数组，第一个元素被隔离

Question

I start with something like

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

what I want is something like:

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

I currently have something

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

Answer 1

You are missing []

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

With a list comprehension

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