acces columns in array of array python code example

Example 1: how get 1st column in all rows of a 2d matrix in python

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

# output
>>> 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)

Example 2: get column or row of matrix array numpy python

test = numpy.array([[1, 2], [3, 4], [5, 6]])
column = test[:,0]
# column == array([1, 3, 5])
row = test[1,:]
# row == array([3, 4])