it-swarm.com.de

So fügen Sie einem pandas dataframe einfach eine Spaltenebene hinzu

nehmen wir an, ich habe einen Datenrahmen, der so aussieht:

df = pd.DataFrame(index=list('abcde'), data={'A': range(5), 'B': range(5)})
 df
Out[92]: 
   A  B
a  0  0
b  1  1
c  2  2
d  3  3
e  4  4

Angenommen, dieser Datenrahmen ist bereits vorhanden, wie kann ich dem Spaltenindex einfach eine Ebene "C" hinzufügen, sodass ich Folgendes erhalte:

 df
Out[92]: 
   A  B
   C  C
a  0  0
b  1  1
c  2  2
d  3  3
e  4  4

Ich sah SO eine Antwort wie diese Python/Pandas: wie kombiniert man zwei Datenrahmen zu einem mit hierarchischem Spaltenindex? aber das setzt andere Datenrahmen zusammen, anstatt eine Spaltenebene hinzuzufügen ein bereits vorhandener Datenrahmen.

-

32
Steven G

Wie von @StevenG selbst vorgeschlagen, eine bessere Antwort:

df.columns = pd.MultiIndex.from_product([df.columns, ['C']])

print(df)
#    A  B
#    C  C
# a  0  0
# b  1  1
# c  2  2
# d  3  3
# e  4  4
45
Romain

Option 1
set_index und T

df.T.set_index(np.repeat('C', df.shape[1]), append=True).T

Option 2
pd.concat, keys und swaplevel

pd.concat([df], axis=1, keys=['C']).swaplevel(0, 1, 1)

enter image description here

15
piRSquared