我有以下数据框
| salesman | north_access | south_access | east_access | west_access |
|---|---|---|---|---|
| A | 1 | 0 | 0 | 1 |
| B | 0 | 1 | 1 | 1 |
| C | 1 | 0 | 1 | 1 |
我想将上面的格式转换为下面的格式
| salesman | direction | access |
|---|---|---|
| A | north | 1 |
| A | south | 0 |
| A | east | 0 |
| A | west | 1 |
| B | north | 0 |
| B | south | 1 |
| B | east | 1 |
| B | west | 1 |
我试着try 拆分和转置功能，但没有得到预期的结果. 有没有人能帮个忙，在代码中做以上的修改，提前谢谢.
另一种解决方案(使用pd.wide_to_long):
df.columns = [f'access_{c.split("_")[0]}' if "_access" in c else c for c in df.columns]
x = pd.wide_to_long(
df, stubnames="access", suffix=r".*", i=["salesman"], j="direction", sep="_"
).reset_index()
print(x)
打印:
salesman direction access
0 A north 1
1 B north 0
2 C north 1
3 A south 0
4 B south 1
5 C south 0
6 A east 0
7 B east 1
8 C east 1
9 A west 1
10 B west 1
11 C west 1