unique values in a column dataframe code example

Example 1: python extract values that have different values in a column

df = pd.DataFrame({'author':['a', 'a', 'b'], 'products':['sr1', 'sr2', 'sr2']}) # Create df

group = df.groupby('author') # Group by author used as index
df2 = group.apply(lambda x: x['subreddit'].unique()) # List of all product per author
df2 = df2.apply(pd.Series) # Split products in multiple columns

Example 2: if df[col].unique()==2

df_dict = dict(zip([i for i in df.columns] , [pd.DataFrame(df[i].unique(), columns=[i]) for i in df.columns]))