python - 操作数据框字典中的元素

Question

我正在尝试遍历数据框字典。对于字典中的每个 df，我试图将两列附加到一个列表，执行一些操作，然后将该列表附加到一个单独的主数据框。

我如何尝试访问字典的每个元素:

for k in range(len(dict_of_df)):
    l = dict_of_df[['Name1', 'Name2']][k].to_list()

然后我试图从这个列表中删除任何重复项，并将剩余项目的组合附加到主 df 的两个单独的列中。

例如，从一开始就给定以下数据框:

Name1    Name2    Color    Size
 Jim      Bob      R         L
 Sam      Jim      G         S

列 Name1 & Name2 将被发送到如下列表:

 l
Jim
Bob
Sam
Jim

删除重复项会留下:

 l
Jim
Bob
Sam

将这些项目的组合插入主 df 将产生:

 Name1   Name2
  Jim     Bob
  Jim     Sam
  Sam     Bob

Answer 1

您可以使用 itertools.combinations，一次取两个。展平您的姓名列，然后在进行组合之前删除重复项。

# pandas >= 0.24
from itertools import combinations

names = pd.Series(
    df[['Name1', 'Name2']].to_numpy().ravel()).drop_duplicates()
pd.DataFrame(combinations(names, 2), columns=['name1', 'name2'])

  name1 name2
0   Jim   Bob
1   Jim   Sam
2   Bob   Sam

---

# pandas < 0.24
# .to_numpy() -> .values
# combinations(...) -> list(combinations(...)) # iterators not supported
names = pd.Series(df[['Name1', 'Name2']].values.ravel()).drop_duplicates()
pd.DataFrame([*combinations(names, 2)], columns=['name1', 'name2'])

  name1 name2
0   Jim   Bob
1   Jim   Sam
2   Bob   Sam