python - 如何从其他列字符串创建新列？

Question

输入 df:

A Jan.S1 Jan.S2 Feb.S1 Feb.S2
x 1      2      3      4
y 6      7      8      9

输出 df:

A  month S1 S2 
x  Jan   1  2
x  Feb   3  4
y  Jan   6  7
y  Feb   8  9

如何让输入变成输出格式？

Answer 1

如果您只有在“.”之后有一个数值你可以使用 pandas.wide_to_long .

由于情况并非如此，您可以使用 MultiIndex 和 stack 进行手动 reshape :

out = (df
 .set_index('A')
 .set_axis(df.columns.drop('A').str.split('.', expand=True), axis=1)
 .stack(0).rename_axis(['A', 'month'])
 .reset_index()
)

输出:

   A month  S1  S2
0  x   Feb   3   4
1  x   Jan   1   2
2  y   Feb   8   9
3  y   Jan   6   7