从 pandas dataframe 到多维 numpy 数组以兼容 Z2C39BC19B761AC36DC046245D1

Question

我有一个 pandas dataframe 其中一列包含列表：

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)
)

我想将此数据作为 3D 数组传递到 tensorflow 中，但首先我需要能够从中获得正确的形状。

非常感谢！

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)