python - 使用具有相同列的数据创建新列

Question

我有与此类似的 DataFrame。如何添加具有在列之一中具有相同值的行名称的新列？例如:
有这个:

  name  building 
  a     blue
  b     white
  c     blue
  d     red
  e     blue
  f     red

如何得到这个？

  name  building  in_building_with
  a     blue      [c, e]
  b     white     []
  c     blue      [a, e]
  d     red       [f]
  e     blue      [a, c]
  f     red       [d]

Answer 1

这是我只能想到的方法(最糟糕的):

r = df.groupby('building')['name'].agg(dict)
df['in_building_with'] = df.apply(lambda  x: [r[x['building']][i] for i in (r[x['building']].keys()-[x.name])], axis=1)

df:

name    building    in_building_with
0   a   blue    [c, e]
1   b   white   []
2   c   blue    [a, e]
3   d   red     [f]
4   e   blue    [a, c]
5   f   red     [d]

联系方式:

制作一本字典，它会给出建筑物发生的索引。

building
blue     {0: 'a', 2: 'c', 4: 'e'}
red              {3: 'd', 5: 'f'}
white                    {1: 'b'}
dtype: object

从列表中减去当前建筑物的索引，因为您正在查看它以外的元素以获得外观索引。

r[x['building']].keys()-[x.name]

获取这些索引处的值并将它们放入一个列表中。