Прикрепление метки к вектору изображения

Question

У меня есть список images, который содержит вектор изображения и соответствующую метку (т.е. класс).

images.append([np.array(image_array_to_vector),np.array([label])])

Если я хочу прочитать изображения в X и метки в y Я сделал следующее:

X = np.array([i[0] for i in images])
y = [i[1] for i in images]

Затем я хочу сохранить X в текстовом файле следующим образом:

X_to_text_file = np.savetxt('x.txt', X.reshape(np.shape(X)), fmt='%5f')

Вышеприведенное работает отлично.Я хотел бы внести изменение, прикрепив метку к вектору в X и сохранив ее в текстовом файле.

Я попытался сделать это, например:

X = np.array([[i[0],i[1]] for i in images])

Но, получил следующую ошибку:

TypeError: Mismatch between array dtype ('object') and format specifier ('%5f %5f')

И, при попытке использовать словарь следующим образом:

X = np.array([{i[1]:i[0]} for i in images])

Я получаю эту ошибку:

TypeError: unhashable type: 'numpy.ndarray'

Так, да, как я могу прикрепить метку к вектору и сохранить ее в текстовом файле?

Спасибо.

Answer 1

немного сложно ответить на этот вопрос, не зная, как выглядят исходные данные или как вы хотите извлечь информацию из .txt в будущем, но изменив формат "% 5f" на строковый формат "% s"должен решить вашу проблему:

X_to_text_file = np.savetxt('x.txt', X.reshape(np.shape(X)), fmt='%s')