仅透视列中的某些值

Question

我有这样的dataframes（key_的个数总是等于value_的个数，但是这个数字是事先不知道的）：

    firstid    secondid    key_1    key_2    key_3    ...    key_n    value_1    value_2    value_3    ...    value_n
      1          one         A        B        C       ...    Z        alpha    beta          gamma    ...    omega

我想把它们改造成这种格式：

first_id    second_id    key    value
1             one         A    alpha
1             one         B    beta
1             one         C    gamma
1             one         ...    ...
1             one         Z    omega

通过像这样使用melt...

value_vars = [x for x in df.columns if x.startswith('key_') or x.startswith('value_')]

df = pd.melt(df, id_vars=["firstid",
                            "secondid"],
              value_vars=value_vars)

...我得到这个结果：

   firstid secondid variable  value
0        1      one    key_1      A
1        1      one    key_2      B
2        1      one    key_3      C
3        1      one    key_n      Z
4        1      one  value_1  alpha
5        1      one  value_2   beta
6        1      one  value_3  gamma
7        1      one  value_n   omega

这显然仍然是一个步骤，但我无法弄清楚在融化、枢轴、堆叠等丛林中是哪一个。欢迎任何解释或线索。

Answer 1

我认为你需要：

• 第一个set_index 用于所有没有_ 的列
• 在split的列中创建MultiIndex
• 由stack重塑
• 双重数据清洗reset_index

---

df = df.set_index(['firstid','secondid'])
df.columns = df.columns.str.split('_', expand=True)
df = df.stack().reset_index(level=2, drop=True).reset_index()
print (df)
   firstid secondid key  value
0        1      one   A  alpha
1        1      one   B   beta
2        1      one   C  gamma
3        1      one   Z   zeta