python - 一行 PANDAS 的多个结果

Question

我有一个包含名称和部门的数据框。所有这些名称可以映射到 4 个不同的部门(部门 A、B、C 和 D)。例如，每个名称必须映射到部门 A、B、C、D。在初始数据中，每个名称仅映射到一个部门。在最终的数据集中，每个名称映射到四个不同的部门。我需要一种方法为每个名称插入 3 个附加行，并为其添加 3 个其他部门。

示例:

初始数据:

Name Department
Joe  A
Ellen A
Zach  A

最终数据:

Name Department
Joe  A
Joe  B
Joe  C
Joe  D
Ellen A
Ellen B
Ellen C
Ellen D
Zach  A
Zach  B
Zach  C
Zach  D

我知道这可以在某种 for 循环中完成，但我对 Pandas 很陌生，任何帮助都将非常感激。

Answer 1

您可以尝试将所有部门作为列表分配给每个名称，然后.explode:

df = df.assign(Department=[["A", "B", "C", "D"]] * len(df)).explode(
    "Department"
)
print(df)

打印:

    Name Department
0    Joe          A
0    Joe          B
0    Joe          C
0    Joe          D
1  Ellen          A
1  Ellen          B
1  Ellen          C
1  Ellen          D
2   Zach          A
2   Zach          B
2   Zach          C
2   Zach          D