til/python/converting-dataframe-column-to-numpy-array.md

543 B

This returns a numpy array:

arr = df["cluster"].to_numpy()

This returns a numpy array of unique values:

unique_arr = df["cluster"].unique()

You can also use numpy to get the unique values, although there are differences between the two methods(df.unique doesn't sort while np.unique does):

arr = df["cluster"].to_numpy()
unique_arr = np.unique(arr)