pandas DataFrame Diagonale

Was ist ein effizienter Weg, um die Diagonale eines Quadrats DataFrame. Ich würde erwarten, dass das Ergebnis ein Series mit einem MultiIndex mit zwei Ebenen, die erste ist die index der DataFrame der zweiten Ebene werden die Spalten der DataFrame.

Setup

import pandas as pd
import numpy as np

np.random.seed([3, 1415])
df = pd.DataFrame(np.random.rand(3, 3) * 5,
                  columns = list('abc'),
                  index = list('ABC'),
                  dtype=np.int64
                 )

Das will ich sehen:

print df.stack().loc[[('A', 'a'), ('B', 'b'), ('C', 'c')]]

A  a    2
B  b    2
C  c    3
InformationsquelleAutor piRSquared | 2016-05-18
Schreibe einen Kommentar