arrays - Numpy 数组操作 Y[ ..., 0]

Question

只是一个简单的问题，

我遇到了一个对代码很重要的 NumPy 数组操作，但我无法理解它的作用。

Probs = Y[ ... , 0]

代码供参考Python 代码

任何帮助将不胜感激。

标签： arrayspython-3.xnumpy

Answer 1

在 2d numpy 数组中，这应该为您提供每个嵌套列表的第一个元素，例如：

import numpy as np
Y = np.asarray([[1,2,3,4], [1,2,3,4], [2,3,4,5]])
Y[..., 0]
>>> array([1, 1, 2])