Как мне взять 1D NumPy массив и превратить его в DF с конкретными именами

Question

Я запускаю изображение через автокодер, чтобы вернуть 1D вектор с размером (224, ). Я хочу взять этот 1D вектор и перенести его в (, 224) и поставить префикс имени столбца, например image_0, image_1....image_224. Как мне это сделать?

Я пробовал следующее:

import numpy as np()

test = np.zeros(224,)

test = test.transpose()

Но это оставляет форму как есть. Позже я прочитал в документации, что .transpose не влияет на одномерный массив.

Есть ли простой / чистый способ сделать это?

Answer 1

Как я понимаю, вы хотите сделать что-то вроде этого:

test = np.zeros(224,)
df = pd.DataFrame(test, index=[f"image{int(i)}" for i in range(len(test))])