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

У меня есть эта тестовая таблица в pandas 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, моя таблица содержит больше строк, это всего лишь моментальный снимок.

One Solution collect form web for “Функция для выбора из столбцов pandas df”

Сначала вы можете использовать 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 

Редактировать комментарий:

 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_id DataFrame :

 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 
  • Роллинг idxmax () в python?
  • Как сформировать столбец кортежа из двух столбцов в Pandas
  • Выбор столбца pandas по местоположению
  • как извлечь элемент серии pandas и сравнить его с строками в столбце dataframe
  • Pandas DataFrame: конвертировать WKT в GeoJSON в новый столбец, используя функцию Lambda
  • Как заполнить отсутствующую запись данных Pandas в виде pythonic?
  • Есть ли способ (довольно) распечатать всю Pandas Series / DataFrame?
  • Как удалить только имя индекса, а не содержимое в кадре данных Pindas multiindex
  • Python - лучший язык программирования в мире.