Преобразование Pandas Dataframe в 3D-матрицу ⇐ Python

[Anonymous]

У меня есть pandas.DataFrame, как показано ниже, вы можете прочитать его как pd.DataFrame(data_dict):

Код: Выделить всё

data_dict =
{'Elevation': {0: 0, 1: 0, 2: 0, 3: 0, 4: 0, 5: 1, 6: 1, 7: 1, 8: 1, 9: 1},
'Azimuth': {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 0, 6: 1, 7: 2, 8: 3, 9: 4},
'median': {0: 255,
1: 255,
2: 255,
3: 255,
4: 255,
5: 256,
6: 256,
7: 256,
8: 256,
9: 256},
'count': {0: 255,
1: 255,
2: 255,
3: 255,
4: 255,
5: 250,
6: 250,
7: 250,
8: 250,
9: 250},
'to_drop': {0: 1,
1: 1,
2: 1,
3: 1,
4: 1,
5: 0,
6: 0,
7: 0,
8: 0,
9: 0}}

Я хочу преобразовать его в 3D-матрицу в numpy. Форма трехмерной матрицы будет такой: [Azimuth.nunique(), Elevation.nunique(), 3(Median,count,to_drop)] т. е. [5,2,3].
Я попробовал data.groupby(['Elevation','Azimuth']).apply(lambda x: x.values).reset_index().values, в результате получился массив 10,3. Как получить массив 5,2,3?

Подробнее здесь: https://stackoverflow.com/questions/783 ... mpy-matrix