Как преобразовать 3D-массив в датафрейм - PullRequest
0 голосов
/ 06 сентября 2018

У меня есть массив измерений (40 X 40 X 8064), который соответствует (данные видео X канала X).

Но теперь я хочу преобразовать массив в таком порядке во фрейм данных, чтобы он выглядел как

Index  |  Video  |  Channel_0  |  Channel_1  |  Channel_2  |  ....  |  Channel_39
0      |     0   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]
1      |     0   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]
2      |     0   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]
3      |     0   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]
4      |     0   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]
...............
8063   |     0   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]
......
......
322559 |     39   |[Some Value] |[Some Value] |[Some Value] |  ....  |[Some Value]

1 Ответ

0 голосов
/ 06 сентября 2018
import numpy as np
import pandas as pd

n_videos = 2
n_channels = 3
n_points = 4

# Generate data
A = np.arange(n_videos * n_channels * n_points).reshape(n_videos,
                                                        n_channels,
                                                        n_points)

col_names = ["Channel_{}".format(i) for i in range(n_channels)]
# Need to re-order axis before reshape. 
# Want axis correspoonding to videos to be at first axis.
df = pd.DataFrame(np.rollaxis(A, 2, 1).reshape(-1, n_channels),
                  columns=col_names)
video_idx = np.hstack((np.full((n_points,), i) for i in range(n_videos)))
video_idx = pd.Series(video_idx, name="Video")
df.insert(0, "Video", video_idx)
...