将2D numpy数组拆分成两个1D数组的最简单方法是什么？

Question

如何从数组a中获取数组x和y？

a是一个二维数组，我需要获取x和y，它们是a的列。

a=[[ 0  1]
  [ 2  3]
  [ 4  5]
  [ 6  7]
  [ 8  9]
  [10 11]]

x =[[0]   
    [2]
    [4]
    [6]
    [8]  
    [10]]

y =  [[1]
     [3]
     [4]
     [7]
     [9]
     [11]]

- caveskelton

---

使用切片获取列：x，y = a[:,:1]，a[:,1:] - Psidom

Answer 1

您的数组：

arr=np.array([[0,1],[2,3],[4,5],[6,7],[8,9],[10,11]])

your answer:-

arr1,arr2=np.split(arr,2,axis=1)

- Anurag Dabas

---

我实际需要的是你代码后的 arr1.flatten() 和 arr2.flatten()。我们能否直接从 array 中获取这两个一维数组？ - Aman Kushwaha

Answer 2

试试这个

numpy.split(array, 2, axis=1)

- Tim

---

如果我想要以一维数组的形式得到答案怎么办？ - Khan Juned

假设你有一个名为arr的二维数组......只需使用arr.flatten()，就可以获得一维数组。 - Anurag Dabas

对每个子数组使用flatten吗？不需要。 - alercelik