python - 如何将我的数据框转换为二维列表，并将所有列作为列表

Question

我有一个df，

Name  Class
Sri   A
Ram   B

我的预期输出是

 [["Name","Sri","Ram"],["Class","A","B"]]

我尝试了 df.values.tolist() 但给出了按行列表，我需要按列排列。提前致谢

Answer 1

使用转置:

print (df.values.T.tolist())
[['Sri', 'Ram'], ['A', 'B']]

如果需要，列名也首先转置为 T与 reset_index :

print (df.T.reset_index().values.tolist())
[['Name', 'Sri', 'Ram'], ['Class', 'A', 'B']]

编辑:要删除 NaN 使用列表理解:

print (df)
  Name Class
0  Sri     A
1  Ram   NaN

L = df.values.T.tolist()
print (L)
[['Sri', 'Ram'], ['A', nan]]

L1 = [[i for i in x if pd.notnull(i)] for x in L]
print (L1)
[['Sri', 'Ram'], ['A']]