Как преобразовать массив numpy в фреймворк pandas?

Я хотел бы иметь 3 столбца массива numpy

px[:,:,0]
px[:,:,1]
px[:,:,0]

в pandas Dataframe.

Должен ли я использовать?

df = pd.DataFrame(px, columns=['R', 'G', 'B'])

Спасибо

Уго

Ответы

Ответ 1

Сначала нужно изменить форму массива, попробуйте следующее:

px2 = px.reshape((-1,3))
df = pd.DataFrame({'R':px2[:,0],'G':px2[:,1],'B':px2[:,2]})