¿Cómo se extrae una columna de una matriz multidimensional?

¿Alguien sabe cómo extraer una columna de una matriz multidimensional en Python?

>>> import numpy as np
>>> A = np.array([[1,2,3,4],[5,6,7,8]])

>>> A
array([[1, 2, 3, 4],
    [5, 6, 7, 8]])

>>> A[:,2] # returns the third columm
array([3, 7])

Ver también: “numpy.arange” y “reformar” para asignar memoria

Ejemplo: (Asignación de una matriz con forma de matriz (3×4))

nrows = 3
ncols = 4
my_array = numpy.arange(nrows*ncols, dtype="double")
my_array = my_array.reshape(nrows, ncols)

¿Podría ser que estés usando un matriz numérica? Pitón tiene la formación módulo, pero eso no es compatible con matrices multidimensionales. Las listas normales de Python también son unidimensionales.

Sin embargo, si tiene una lista bidimensional simple como esta:

A = [[1,2,3,4],
     [5,6,7,8]]

entonces puedes extraer una columna como esta:

def column(matrix, i):
    return [row[i] for row in matrix]

Extrayendo la segunda columna (índice 1):

>>> column(A, 1)
[2, 6]

O alternativamente, simplemente:

>>> [row[1] for row in A]
[2, 6]

Si tienes una matriz como

a = [[1, 2], [2, 3], [3, 4]]

Luego extraes la primera columna así:

[row[0] for row in a]

Así que el resultado se ve así:

[1, 2, 3]

¡Échale un vistazo!

a = [[1, 2], [2, 3], [3, 4]]
a2 = zip(*a)
a2[0]

es lo mismo que el anterior, excepto que de alguna manera es más ordenado, el zip hace el trabajo pero requiere matrices individuales como argumentos, la sintaxis *a desempaqueta la matriz multidimensional en argumentos de matriz única

>>> x = arange(20).reshape(4,5)
>>> x array([[ 0,  1,  2,  3,  4],
        [ 5,  6,  7,  8,  9],
        [10, 11, 12, 13, 14],
        [15, 16, 17, 18, 19]])

si quieres la segunda columna puedes usar

>>> x[:, 1]
array([ 1,  6, 11, 16])

Si tiene una matriz bidimensional en Python (no numpy), puede extraer todas las columnas así,

data = [
['a', 1, 2], 
['b', 3, 4], 
['c', 5, 6]
]

columns = list(zip(*data))

print("column[0] = {}".format(columns[0]))
print("column[1] = {}".format(columns[1]))
print("column[2] = {}".format(columns[2]))

Ejecutar este código producirá,

>>> print("column[0] = {}".format(columns[0]))
column[0] = ('a', 'b', 'c')

>>> print("column[1] = {}".format(columns[1]))
column[1] = (1, 3, 5)

>>> print("column[2] = {}".format(columns[2]))
column[2] = (2, 4, 6)

def get_col(arr, col):
    return map(lambda x : x[col], arr)

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

print get_col(a, 3)

La función de mapa en Python es otra forma de hacerlo.