python – 从数组数组中提取数组

概述我有这个数组： arr = np.array([[[ -1., -1., -1., 0., 0., 0.], [ 0.1, 0.1, 0.1, 2., 3., 4.]], # <-- this one [[ -1., -1., -1., 0., 0., -1.], 我有这个数组：

arr = np.array([[[ -1.,-1.,0.,0.],[ 0.1,0.1,2.,3.,4.]],# <-- this one               [[ -1.,-1.],16.,17.,0.1]],# <-- and this one               [[ -1.,4.,5.,6.]],# <-- and this one               [[  0.,[  1.,1.,2.]],9.,10.,11.]]]) # <-- and the last one

我想在每个数组中提取第二个数组,结果如下：

res = [[ 0.1,4.],0.1],6.],2.],11.]]

我希望在一行代码中获得res,我试过这个,但它没有用

arr[:][1] # select the element 1 in each array# I gotarray([[ -1.,-1. ],[  0.1,0.1]])

有谁能解释为什么？

我找到的唯一解决方案是明确指出每个索引(arr [0] [1] …),我不喜欢.

解决方法 这是一个3D数组,您正在尝试选择第二个轴的第二个元素并沿其余轴提取所有元素.所以,它很简单 –

arr[:,1,:]

我们可以跳过列出：对于尾轴,所以它进一步简化为 –

arr[:,1]

样品运行 –

In [360]: arrOut[360]: array([[[ -1.,0. ],4. ]],[[ -1.,6. ]],[[  0.,2. ]],11. ]]])In [361]: arr[:,1]Out[361]: array([[  0.1,4. ],6. ],2. ],11. ]])

总结

以上是内存溢出为你收集整理的python – 从数组数组中提取数组全部内容，希望文章能够帮你解决python – 从数组数组中提取数组所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。