h5py - преобразовать dtype из набора данных из int / float в bool?

Question

Итак, чтобы дать вам представление о структуре моего набора данных, когда я делаю print (hf ["y_train"] [0]) Я получаю это в результате: [3.2355e-05 3.3180e-05 1.0000e + 00]

Значения в третьем столбце равны либо 1,0, либо 0,0; я пытаюсь преобразовать 1,0 с в Истину и 0,0 с в Ложь.

Я уже пробовал множество вещей, но безрезультатно.

Answer 1

Это не похоже на проблему HDF5 / h5py. Кажется, что данные были сохранены как float и читаются как float, как и должно быть.

Чтобы преобразовать в bool после чтения, вы можете попробовать np.isclose со стандартными уровнями допуска:

A = np.array([3.2355e-05, 3.3180e-05, 1.0000e+00])

res = np.isclose(A, 1)

array([False, False,  True])