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.
-
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
Option 1set_index und T
df.T.set_index(np.repeat('C', df.shape[1]), append=True).T
Option 2pd.concat, keys und swaplevel
pd.concat([df], axis=1, keys=['C']).swaplevel(0, 1, 1)
