Функция выбора из столбцов панды ДФ

Question

У меня есть тестовый стол в панд dataframe

Leaf_category_id session_id product_id 
0    111   1   987 
3    111   4   987 
4    111   1   741 
1    222   2   654 
2    333   3   321 

, что я хочу

for leaf_category_id 111: 

результат должен быть.

session_id product_id 
1   987,741 
4   987 

Аналогично может я определить функцию, которая делает то же самое для всех leaf_category идентификаторов, моя таблица содержит несколько строк, это просто снимок этого.

Answer 1

Вы можете использовать boolean indexing первый, а затем groupby с применить join:

df = pd.DataFrame({'Leaf_category_id':[111,111,111,222,333], 
        'session_id':[1,4,1,2,3], 
        'product_id':[987,987,741,654,321]}, 
        columns =['Leaf_category_id','session_id','product_id']) 

print (df)     
    Leaf_category_id session_id product_id 
0    111   1   987 
1    111   4   987 
2    111   1   741 
3    222   2   654 
4    333   3   321 


print (df[df.Leaf_category_id == 111] 
      .groupby('session_id')['product_id'] 
      .apply(lambda x: ','.join(x.astype(str)))) 
session_id 
1 987,741 
4  987 
Name: product_id, dtype: object 

EDIT замечанием:

print (df.groupby(['Leaf_category_id','session_id'])['product_id'] 
     .apply(lambda x: ','.join(x.astype(str))) 
     .reset_index()) 
    Leaf_category_id session_id product_id 
0    111   1 987,741 
1    111   4  987 
2    222   2  654 
3    333   3  321 

Или, если необходимо для каждого уникального значения в Leaf_category_idDataFrame:

for i in df.Leaf_category_id.unique(): 
    print (df[df.Leaf_category_id == i] \ 
       .groupby('session_id')['product_id'] \ 
       .apply(lambda x: ','.join(x.astype(str))) \ 
       .reset_index()) 

    session_id product_id 
0   1 987,741 
1   4  987 
    session_id product_id 
0   2  654 
    session_id product_id 
0   3  321