Понимание списка для создания вектора в два раза больше квадрата столбца

Question

Мне нужно написать понимание списка, чтобы создать вектор, в два раза больше квадрата среднего столбца матрицы.(Моя матрица x = [[1,2,3],[4,5,6],[7,8,9]].) Проблема в том, что я знаю, как извлечь средний столбец, НО я не знаю, как его вычеркнуть или удвоить квадрат.Любая помощь будет принята с благодарностью (... еще учусь, но стараюсь изо всех сил)!

x = np.array([[1,2,3],[4,5,6],[7,8,9]])
print(x)

z = [b[1] for b in x]
print(z)

Answer 1

Чтобы создать вектор, в два раза больше квадрата столбца:

import numpy as np

x = np.array([[1,2,3],[4,5,6],[7,8,9]])
print(x)

с пониманием списка: (не рекомендуется)

z = [2*b[1]**2 for b in x]
print(z)

На выходе получается список python:

[8, 50, 128]

с использованием индексации с нуля: (рекомендуется)

дополнительная информация здесь

z = 2 * x[:,1] ** 2 
print(z)

Выход numpy array :

[  8  50 128]