1

I have an Array of Dimensions (40 X 40 X 8064) which Corresponds to (Video X Channel X Data).

But Now I want to Convert the array in such an order to a data frame that it looks like 

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]
Sabby
  • 9
  • 6

1 Answers1

0
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)
lightalchemist
  • 10,031
  • 4
  • 47
  • 55