grab first 2 collumns from np array code example
Example 1: extract column numpy array python
# Extract/ Slice only the 3rd column from the numpy array
print(a2[:,[2]])
Example 2: how to get the first few lines of an ndarray
In [11]: a[:,:2]
Out[11]:
array([[-0.57098887, -0.4274751 ],
[-0.22279713, -0.51723555],
[ 0.67492385, -0.69294472],
[ 0.41086611, 0.26374238]])