In [1]:import pandas as pdIn [2]:import numpy as npIn [3]: company=["A","B","C"]...:...: data=pd.DataFrame({...:"company":[company[x]for x in np.random.randint(0,len(company),10)],...:"salary":np.random.randint(5,50,10),...:"age":np.random.randint(15,50,10)...:}...:)In [4]: company
Out[4]:['A','B','C']In [5]: data
Out[5]: company salary age
0 C 45181 B 43332 A 19213 B 39214 A 49285 A 14