Добавляйте в панды числовые массивы в виде строк со строкой в ​​качестве индекса

Question

У меня есть список строк с именами изображений и массив номеров с особенностями формы (4096,) для каждого изображения.

например, список имен:

img_list = ['img1', 'img2', 'img3']

features np.array

[array([0.        , 0.04688909, 0.05700445, ..., 0.        , 0.        ,
        0.        ], dtype=float32),
 array([0.        , 0.04688909, 0.05700445, ..., 0.        , 0.        ,
        0.        ], dtype=float32),
 array([0.03546982, 0.00310931, 0.        , ..., 0.        , 0.01513313,
        0.        ], dtype=float32)]

я хочу добавить их в фрейм данных pandas, где индексом будет имя img, а функции будут представлены как 1 по столбцу, что-то вроде этого:

    f_1 f_2 f_3 ... f_4094  f_4095  f_4096
img 

Индекс из списка imgs такой же, как и у массива features, поэтому у img [0] есть массив features [0] и т. Д.

Как мне удается объединить их в одну форму данных Nх 4096?

Спасибо!

Answer 1

IIUC:

pd.DataFrame(features, img_list).add_prefix('f_')

MCVE

img_list = ['img1', 'img2', 'img3']
features = [np.array([1, 2, 3]), np.array([4, 5, 6]), np.array([7, 8, 9])]

df = pd.DataFrame(features, img_list).add_prefix('f_')
df

      f_0  f_1  f_2
img1    1    2    3
img2    4    5    6
img3    7    8    9