python, pandy: InvalidIndexError podczas tworzenia dataframe

I zostały odkrywania zestawu danych tytaniczny . Próbuję utworzyć dataframeco będzie miało w wieku od ludzi, którzy przeżyli zatonięcie Titanica i tych, którzy nie w dwóch oddzielnych kolumnach.

    train = pd.read_csv('train.csv')
    test = pd.read_csv('test.csv')    
    whole = pd.concat([train, test])
    df = pd.DataFrame({'survived': whole['Age'][whole['Survived'] == 1],
                       'died': whole['Age'][whole['Survived'] == 0]})

Ale ja dostaję ten błąd

pandas.indexes.base.InvalidIndexError: Ponowne indeksowanie ważne tylko przy wyjątkowo cennych obiektów Index

Co ja robię źle?