numpy select columns code example

Example 1: extract column numpy array python

# Extract/ Slice only the 3rd column from the numpy array
print(a2[:,[2]])

Example 2: numpy how to slice individual columns

X = data[:, [1, 9]]