Выбор столбца из массива в Python

Question

В попытке выбрать первый (если считать от нуля) столбца в 2D массив 4x4, я написал следующий скрипт:

import numpy 
a4x4=[list(range(4*i,4*(i+1))) for i in list(range(4))] 
print(a4x4) 
print(a4x4[:,1]) 

Массив, кажется, в порядке:

[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15]] 

но вместо

[1, 5, 9, 13]

для второй печати, я получаю эту ошибку:

TypeError: list indices must be integers, not tuple

Зачем возникает эта ошибка, что происходит не так?

Answer 1

Вы импортируете numpy, но не используете его. У вас есть список списков, и Python не поддерживает многомерную нарезку для этого (т. Е. Вам понадобится [a4x4[i][1] for i in range(4)], чтобы получить ожидаемый результат, но на самом деле вы должны использовать numpy). Вот пример:

import numpy 
a4x4=numpy.array([list(range(4*i,4*(i+1))) for i in list(range(4))]) 
print(a4x4) 
print(a4x4[:,1]) 

Кстати, в NumPy вы также можете создать массив, который вы хотите напрямую, как это:

numpy.arange(4*4).reshape((4,4)) 

(А также в Python один не нуждается в list вызовов У меня выше, я просто стараюсь, чтобы код как можно более похожий на ваш, чтобы увидеть ключевую вещь, которая преобразует список списков в массив numpy.)

Answer 2

Вы можете создать результат, который хотите использовать список - так же, как вы создали оригинал 4x4:

a4x4=[list(range(4*i,4*(i+1))) for i in list(range(4))] 
print([a4x4[i][1] for i in range(4)]) 

Кроме того, вы можете упростить вашу логику немного, бросая из функции list:

a4x4 = [range(4*i,4*(i+1)) for i in range(4)] 
print([a4x4[i][1] for i in range(4)])