Как извлечь элементы из матрицы в Python

Question

Я новичок в Python, и я сделал матрицу из сотен строк и всего 2 столбца.

Я построил такую ​​матрицу, используя numpy:

FILE = np.array(
    [
[11,12],
[21,22],
[31,32],
[41,42]
[51,52]
......
    ])

Надеюсь, это правильно.

Допустим, я хочу извлечь только первые 3 элемента из первого столбца (это 11, 21 и 31): как это проще всего сделать?

Есть ли для него какая-нибудь numpy утилита?

Answer 1

Используйте это, чтобы получить первые три строки столбца 0:

FILE[0:3, 0]

Answer 2

Здесь вы можете использовать нарезку. Итак, 0:3 или, точнее, :3 - это первые три строки, затем следующая 0 относится к первому столбцу.

>>> FILE[0:3, 0]
array([11, 21, 31])