将表格 pandas DataFrame 转换为嵌套的 pandas DataFrame

Question

假设我有一个像这样的简单 pd.DataFrame：

d = {'col1': [1, 20], 'col2': [3, 40], 'col3': [5, 50]}

df = pd.DataFrame(data=d)


df
   col1  col2  col4
0     1     3     5
1    20    40    60

有没有办法将其转换为 nasted pandas Dataframe (df_new) ，例如当我调用 df_new.values[0] 时将其作为输出：

array(
[0    1
 1    3
 2    5
Length: 3, dtype: int], dtype=object)

Answer 1

我仍然不认为我理解确切的要求，但这里有一些东西：

获得所需output 的一种方法是：

>>> pd.Series(df.T[0].values)

0    1
1    3
2    5
dtype: int64

如果您想将这些作为二维数组：

>>> np.array(pd.DataFrame(df.T[0].values).reset_index())

array([[0, 1],
       [1, 3],
       [2, 5]])

>>> np.array(pd.DataFrame(df.T[1].values).reset_index())

array([[ 0, 20],
       [ 1, 40],
       [ 2, 50]])