如何分解 np 数组中的维度？

Question

我有一个 numpy 数组：

arr=[[1]
[2]
[3]]

但我需要这个：

arr2= [1 2 3]

你会怎么做这个？

我试着这样做：

arr2=np.empty(1)
for i in range(len(arr)):
arr2=np.append(arr2, arr[i][0])

但它在 arr2 的开头给了我一个额外的、更小的数字，我不明白为什么。

Answer 1

尝试切片：

arr = np.array([[1], [2], [3]])
arr[:, 0]
>>> array([1, 2, 3])

语法

[:, 0]

表示对于每个子数组，获取第一个元素。