human
Give me a Python function to call nth column of array in Python.
assistant
Here is a Python function to call nth column of array in Python

import numpy as np

def nth_column(array, n):
    """Return the nth column of an array."""
    return array[:, n]

# Enter the array
array = np.array([[1, 2, 3], [4, 5, 6]])

# Calling the function
print((nth_column(array, 1)))

#Example Output
# [2 5]

#Remember to inspect code for bugs.