Как создать новый столбец pandas, представляющий собой список всех значений в диапазоне индексов, не включая значение строки - PullRequest
Новая
       16

Как создать новый столбец pandas, представляющий собой список всех значений в диапазоне индексов, не включая значение строки

0 голосов
/ 29 мая 2020

Мне было интересно, можно ли создать новый столбец в кадре данных pandas, который представляет собой список всех значений, НЕ включая значение самой строки. Например, в приведенном ниже df у меня для первой строки в столбцах есть значения «list» [b, c] и значение самой строки «a». Возможно ли это сделать по индексу?

Я пробовал это, но он возвращает список всех значений, объединенных по индексу: 

import pandas as pd 
d = {'index': [1, 1, 1, 2, 2, 3], 'col1': ['a', 'b', 'c', 'd', 'e, f', 'g']}
df = pd.DataFrame(d)
df = df.groupby("index")["col1"].apply(list)

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

enter image description here

Спасибо за любую помощь !!

1 Ответ

1 голос
/ 29 мая 2020

Мы можем сделать explode с groupby, создать целое list в каждом index, затем выполнить set sub 

df['l']=df.col1.str.split(',')
df['new']=df.explode('l').groupby('index')['l'].agg(list).reindex(df['index']).tolist()
df['List']=(df.new.apply(set)-df['l'].apply(set)).apply(list)
df.loc[~df.List.astype(bool),'List']=df.l
df
   index  col1        l         new     List
0      1     a      [a]   [a, b, c]   [c, b]
1      1     b      [b]   [a, b, c]   [a, c]
2      1     c      [c]   [a, b, c]   [a, b]
3      2     d      [d]  [d, e,  f]  [e,  f]
4      2  e, f  [e,  f]  [d, e,  f]      [d]
5      3     g      [g]         [g]      [g]

Обновить 

l=[]
... for x , y in zip(df.l,df.new):
...     x=x.copy()
...     y=y.copy()
...     for i in x:
...         if i in y:
...             y.remove(i)
...     l.append(y)
... 
l
[['b', 'c'], ['a', 'c'], ['a', 'b'], ['e', ' f'], ['d'], []]
df['List']=l
...