Numpy Операция массива Y [..., 0]

Question

Просто быстрый вопрос,

Я наткнулся на операцию массива NumPy, которая важна для кода, и я не могу понять, что это делает.

Probs = Y[ ... , 0]

Код для справки Python код

Любая помощь будет оценена.

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])