从pandas DataFrame列标题获取列表

问题描述

你可以执行以下操作以列表形式获取值:

  1. list(my_dataframe.columns.values)

你也可以简单地使用:(如Ed Chum的答案所示):

  1. list(my_dataframe)

解决方法

我想从pandas DataFrame获取列标题的列表。DataFrame来自用户输入,所以我不知道会有多少列或它们将被称为什么。

例如,如果给我这样的DataFrame

  1. >>> my_dataframe
  2. y gdp cap
  3. 0 1 2 5
  4. 1 2 3 9
  5. 2 8 7 2
  6. 3 3 4 7
  7. 4 6 7 7
  8. 5 4 8 3
  9. 6 8 2 8
  10. 7 9 9 10
  11. 8 6 6 4
  12. 9 10 10 7

我想要一个这样的列表:

  1. >>> header_list
  2. ['y','gdp','cap']

猜你在找的技术问答相关文章

如何检查配对的蓝牙设备是打印机还是扫描仪(Android)
是否允许实体正文进行HTTP DELETE请求?
如何将ZipInputStream转换为InputStream?
java.util.logging Java 8中的变量
PowerMockito.doReturn返回null
Java中的RESTful调用
Swing / Java:如何正确使用getText和setText字符串
特殊字符和重音字符
Android Studio中的ndk.dir错误
错误“找不到主类”