Столбец с кадром в списке строк (с групповым)

Question

У меня есть фрейм данных, и я хочу получить один из его столбцов в виде списка строк, чтобы из чего-то вроде:

df = pd.DataFrame({'customer':['a','a','a','b','b'],
         'location':['1','2','3','4','5']})

я мог получить фрейм данных, например:

a  ['1','2','3']
b  ['4','5']

, где один столбец является клиентом, а другой - списком строк их местоположения.

Я пробовал df.astype (str) .values.tolist (), но не могу сгруппироваться вДля получения списка на одного клиента.

Answer 1

Просто используйте

df.groupby('customer').location.unique()
Out[58]: 
customer
a    [1, 2, 3]
b       [4, 5]
Name: location, dtype: object

Это тип string, просто не отображается цитата

df.groupby('customer').location.unique()[0][0]
Out[61]: '1'

Также вы должны знать, что ввод строки в списке доза не показывает цитату в пандах'объект

pd.Series([['1','2']])
Out[64]: 
0    [1, 2]
dtype: object