Question

假设我有以下数组

test=np.asarray([
    [[ 0.26139668,  0.271985  ,  0.89647514,  0.10352486],
    [ 0.30488212,  0.31384717,  0.84089744,  0.15910256],
    [ 0.32112577,  0.32958562,  0.79216727,  0.20783273]],
    [[ 0.6017121 ,  0.60530397,  0.49432092,  0.50567908],
    [ 0.61610247,  0.61954059,  0.49649699,  0.50350301],
    [ 0.63149809,  0.63477652,  0.49945702,  0.50054298]],
    [[ 0.6017121 ,  0.60530397,  0.49432092,  0.50567908],
    [ 0.61610247,  0.61954059,  0.49649699,  0.50350301],
    [ 0.63149809,  0.63477652,  0.49945702,  0.50054298]],
    [[ 0.6017121 ,  0.60530397,  0.49432092,  0.50567908],
    [ 0.61610247,  0.61954059,  0.49649699,  0.50350301],
    [ 0.63149809,  0.63477652,  0.49945702,  0.50054298]]])

现在我要查找的结果位于test的每个元素的第一列。所以基本上我需要test[0][:,0]和test[1][:,0] ...最后需要test[3][:,0]。

我确信这可以在没有for循环的情况下完成吗？

由于符号[:]应该表示从开始到结束的所有元素，我甚至尝试test[:][:,0]。那没用。 test[:,0][:,0]也不是更好。

那么，for循环的替代方法是什么？

结果应为：

    [0.26139668, 0.30488212, 0.32112577, 0.6017121, 0.61610247, 0.63149809,
 0.6017121, 0.61610247, 0.63149809,0.6017121, 0.61610247, 0.63149809]

或类似的东西。（我只需要整个数组中的那些数字）。

Answer 1

你有一个3D数组，所以只需将0传递给第三个轴：

In [9]: test[:, :, 0]
Out[9]: 
array([[ 0.26139668,  0.30488212,  0.32112577],
       [ 0.6017121 ,  0.61610247,  0.63149809],
       [ 0.6017121 ,  0.61610247,  0.63149809],
       [ 0.6017121 ,  0.61610247,  0.63149809]])

如果你想让它们全部在一个数组中，你只需要ravel切片：

In [11]: test[:, :, 0].ravel()
Out[11]: 
array([ 0.26139668,  0.30488212,  0.32112577,  0.6017121 ,  0.61610247,
        0.63149809,  0.6017121 ,  0.61610247,  0.63149809,  0.6017121 ,
        0.61610247,  0.63149809])

从python中的嵌套数组访问元素

1 个答案: