熊猫多索引 从现有索引字符串

Question

我得到了这样的数据框：

df = pd.DataFrame([[1,2,3,4,5,6], [7,8,9,10,11,12]],columns=["a123X","a123Y","a123Z","b123X", "b123Y", "b123Z"])
a123X  a123Y  a123Z  b123X  b123Y  b123Z
0      1      2      3      4      5      6
1      7      8      9     10     11     12

我想使 X Y Z 成为一个多索引。目前我使用这样的解决方法：

df.columns = (df.columns.str[:-1] + "_" + df.columns.str[-1]).str.split("_", expand=True)
a123       b123        
X  Y  Z    X   Y   Z
0    1  2  3    4   5   6
1    7  8  9   10  11  12

有没有办法使这更容易，它将始终是最后一个字母，这将是索引中的第二级。

Answer 1

重新分配嵌套列表：

df.columns = [df.columns.str[:-1],  df.columns.str[-1]]
print (df)
a123       b123        
X  Y  Z    X   Y   Z
0    1  2  3    4   5   6
1    7  8  9   10  11  12