python - 如何根据条件将python数组拆分为不同的数组？

Question

我有一个这样的python数组：

array([[0.34201428, 0.46875536, 0.37900415, 0.4906195 ],
       [0.58203477, 0.35279346, 0.61418074, 0.37601328],
       [0.3388086 , 0.21167754, 0.37330517, 0.2436498 ],
       [0.57343255, 0.34535545, 0.62878576, 0.38982747]],dtype=float32)

我想根据第零列将数组拆分为不同的簇。最终的输出应该是这样的：第一个数组为，

array([[0.34201428, 0.46875536, 0.37900415, 0.4906195 ],
       [0.3388086 , 0.21167754, 0.37330517, 0.2436498 ]])

第二个数组为，

array([[0.58203477, 0.35279346, 0.61418074, 0.37601328],
       [0.57343255, 0.34535545, 0.62878576, 0.38982747]])

谁能帮我吗。谢谢！！

标签： pythonarrayslist