python:如何执行特定的df操作

Question

这是我的原始Q

python如何在数据帧中执行以下操作

我想检查一下如何从输出开始并返回原始df。

Answer 1

对数字和字符串使用Series.str.extract到2个新列，然后通过DataFrame.pivot进行整形，最后进行一些数据处理以匹配输入：

df[['Year','col']] = df['Year'].str.extract('(d+)([A-Z])')
df = (df.pivot('Year','col','Val')
.rename(columns={'A':'Tval1','B':'Tval2'})
.reset_index().rename_axis(None, axis=1))
df['Year'] +='A'
print (df)
Year  Tval1  Tval2
0   1A      1     34
1   2A      9     56
2   3A      8     67
3   4A      1     78
4   5A      6     89