От pandas массива данных до многомерного массива numpy для совместимости с тензорным потоком

Question

У меня есть pandas фрейм данных, где один из столбцов содержит списки:

import pandas as pd
import numpy as np
(
   pd.DataFrame({
      "x": [[1, 2], [3, 4], [5, 6]]
   )}
   .assign(x = lambda data: data.x.apply(np.array))  # convert lists into numpy arrays
   .to_numpy()
   .shape  # returns (3, 1) when I was hoping for a (3,1,2)
)

Я хотел бы передать эти данные в тензорный поток в виде трехмерного массива, но сначала мне нужно иметь возможность получить правильная форма из этого.

Большое спасибо!

Answer 1

Вы можете получить его как:

import pandas as pd
import numpy as np
npArr = np.array(pd.DataFrame({"x": [[1, 2], [3, 4], [5, 6]],
                               "y": [[1, 2], [3, 4], [5, 6]]}).values.tolist())
print(npArr.shape)