python - 不使用切片的 Numpy 索引

Question

假设我有一个数组 A，其形状为 (4,3,2)。无论如何，是否可以在不使用切片语法的情况下表示 A[:,1,1] 或 A[3,:,0] 之类的内容？

假设我有一个形状为 (4,3,2) 的可能性表 p(A,B,C)。给定任意两个变量的两种状态，我需要知道另一个变量的可能性表。但我不知道我会提前知道哪两个变量的状态。所以我不能使用切片语法。

pABC=np.array([
    [[ 0.9 ,  0.3 ],
    [ 0.8 ,  0.4 ],
    [ 0.1 ,  0.01]],
   [[ 0.08,  0.05],
    [ 0.17,  0.05],
    [ 0.01,  0.01]],
   [[ 0.01,  0.05],
    [ 0.01,  0.15],
    [ 0.87,  0.97]],
   [[ 0.01,  0.6 ],
    [ 0.02,  0.4 ],
    [ 0.02,  0.01]]])

有时我需要索引[:,0,1]，有时我需要索引[2,:,0]。我需要为此索引操作编写一个函数。所以我不能在我的函数中使用切片，因为我不知道我需要切片哪个轴。

谢谢!

Answer 1

感谢@cel。事实证明 slice() 正是我想要的。

numpy 中的“:”是切片(无)。所以 A[1,:,2] 等于 A[1,slice(None),2]。

然后我可以按照我想要的任何维度进行切片。

玩得开心!