给出以下数据框:
d2=pd.DataFrame({'Item':['y','y','z','x'],
'other':['aa','bb','cc','dd']})
d2
Item other
0 y aa
1 y bb
2 z cc
3 x dd
我想在顶部添加一行,然后将其用作multiIndexed标头的第1级。我无法始终预测数据帧将包含多少列,因此新行应允许(即随机字符或数字正常)。 我正在寻找这样的事情:
Item other
A B
0 y aa
1 y bb
2 z cc
3 x dd
但同样,列数会有所不同,无法预测。
提前致谢!
答案 0 :(得分:2)
我认为您可以先按shape
查找列数,然后按range
创建列表。最后创建MultiIndex.from_tuples
。
print (d2.shape[1])
2
print (range(d2.shape[1]))
range(0, 2)
cols = list(zip(d2.columns, range(d2.shape[1])))
print (cols)
[('Item', 0), ('other', 1)]
d2.columns = pd.MultiIndex.from_tuples(cols)
print (d2)
Item other
0 1
0 y aa
1 y bb
2 z cc
3 x dd
如果您需要字母列,并且列数少于26
,请使用:
import string
print (list(string.ascii_uppercase))
['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
print (d2.shape[1])
2
print (list(string.ascii_uppercase)[:d2.shape[1]])
['A', 'B']
cols = list(zip(d2.columns, list(string.ascii_uppercase)[:d2.shape[1]]))
print (cols)
[('Item', 'A'), ('other', 'B')]
d2.columns = pd.MultiIndex.from_tuples(cols)
print (d2)
Item other
A B
0 y aa
1 y bb
2 z cc
3 x dd