python - 如何将 pandas.core.frame.DataFrame 转换为列表？

Question

这是我的 csv 文件中的 。这是因为我想将 header C 中具有“快乐”值的所有 header A 分组

print(df.loc[df["Header C"] == "happy"])

这是输出:

    Header A    Header B    Header C   Header D    Header E
0     rowa.a      rowa.c       happy     rowa.d      rowa.e
3     rowc.a      rowc.c       happy     rowd.d      rowc.e
5     rowe.a      rowe.c       happy     rowe.d      rowe.e

有什么方法可以将输出 header 放入列表并将其存储在变量中吗？

与行相同，我也希望它创建一个列表并将其存储在变量中。

(像这样):

a = ["Header A", "Header B", "Header C", "Header D", "Header E"]
n = ["rowa.a", "rowa.c", "happy", "rowa.d", "rowa.e"]
o = ["rowc.a", "rowc.c", "happy", "rowc.d", "rowc.e"]

我想要的最终输出是一个漂亮的

from prettytable import PrettyTable
x = PrettyTable()
x.field_names = a
x.add_row(n)
x.add_row(o)
print(x)

及其输出:

+----------+----------+----------+----------+----------+
| Header A | Header B | Header C | Header D | Header E |
+----------+----------+----------+----------+----------+
|  rowa.a  |  rowa.c  |  happy   |  rowa.d  |  rowa.e  |
+----------+----------+----------+----------+----------+

非常感谢!

Answer 1

嘿，您可以将数据框转换回字典，然后访问您描述的所有列表:

df.to_dict('split')