按名称从数据框返回列

Question

我有给定列名称的数据框，并且我想返回一个具有指定名称的列：

name_of_column = 'name1'  # string variable

我试图用这个：

dataframe.iloc[:, name_of_column]

但是没有用。我该怎么办？

Answer 1

您可以这样做：

dataframe[column_name]

将选择该列。

iloc()方法通过索引在熊猫中查找项目。

更多示例，您可以在Pandas Indexing and Selecting Data中找到选择数据

Answer 2

使用loc代替iloc，您的语法将起作用。 iloc用于通过整数位置进行索引（这就是i的代表），而loc用于通过 label 进行索引。因此，您可以使用：

dataframe.loc[:, name_of_column]

话虽如此，检索系列的更常用方法是直接使用__getitem__：

dataframe[name_of_column]