Question

在R中，当您需要根据您可以执行的列的名称检索列索引时

idx <- which(names(my_data)==my_colum_name)

有没有办法对pandas数据帧做同样的事情？

Answer 1

当然，您可以使用.get_loc()：

In [45]: df = DataFrame({"pear": [1,2,3], "apple": [2,3,4], "orange": [3,4,5]})

In [46]: df.columns
Out[46]: Index([apple, orange, pear], dtype=object)

In [47]: df.columns.get_loc("pear")
Out[47]: 2

虽然说实话，我自己并不经常需要这个。通常按名称访问可以实现我想要的内容（df["pear"]，df[["apple", "orange"]]或df.columns.isin(["orange", "pear"])），但我绝对可以看到您需要索引编号的情况。

Answer 2

这是通过列表理解的解决方案。 cols是获取索引的列的列表：

[df.columns.get_loc(c) for c in cols if c in df]

Answer 3

DSM的解决方案有效，但如果您希望直接等效于which，则可以执行(df.columns == name).nonzero()

Answer 4

当您可能希望找到多个列匹配时，可以使用使用searchsorted method的矢量化解决方案。因此，使用df作为数据帧并使用query_cols作为要搜索的列名，实现将是 -

def column_index(df, query_cols):
    cols = df.columns.values
    sidx = np.argsort(cols)
    return sidx[np.searchsorted(cols,query_cols,sorter=sidx)]

示例运行 -

In [162]: df
Out[162]: 
   apple  banana  pear  orange  peach
0      8       3     4       4      2
1      4       4     3       0      1
2      1       2     6       8      1

In [163]: column_index(df, ['peach', 'banana', 'apple'])
Out[163]: array([4, 1, 0])

Answer 5

如果您想要列名称中的列名称（与OP问题相反），您可以使用：

>>> df.columns.get_values()[location]

使用@DSM示例：

>>> df = DataFrame({"pear": [1,2,3], "apple": [2,3,4], "orange": [3,4,5]})

>>> df.columns

Index(['apple', 'orange', 'pear'], dtype='object')

>>> df.columns.get_values()[1]

'orange'

另一种方式：

df.iloc[:,1].name

Answer 6

要返回多个列索引，如果您具有唯一标签，建议使用pandas.Index方法get_indexer：

df = pd.DataFrame({"pear": [1, 2, 3], "apple": [2, 3, 4], "orange": [3, 4, 5]})
df.columns.get_indexer(['pear', 'apple'])
# Out: array([0, 1], dtype=int64)

如果索引中具有非唯一标签（列仅支持唯一标签）get_indexer_for。它的参数与get_indeder相同：

df = pd.DataFrame(
    {"pear": [1, 2, 3], "apple": [2, 3, 4], "orange": [3, 4, 5]}, 
    index=[0, 1, 1])
df.index.get_indexer_for([0, 1])
# Out: array([0, 1, 2], dtype=int64)

这两个方法还支持非精确索引，带有f.i。对于采用公差的最接近值的浮点值。如果两个索引到指定标签的距离相同或重复，则选择索引值较大的索引：

df = pd.DataFrame(
    {"pear": [1, 2, 3], "apple": [2, 3, 4], "orange": [3, 4, 5]},
    index=[0, .9, 1.1])
df.index.get_indexer([0, 1])
# array([ 0, -1], dtype=int64)

Answer 7

这如何：

df = DataFrame({"pear": [1,2,3], "apple": [2,3,4], "orange": [3,4,5]})
out = np.argwhere(df.columns.isin(['apple', 'orange'])).ravel()
print(out)
[1 2]

Answer 8

要稍微修改 DSM 的答案，根据当前版本的 Pandas (1.1.5) 中的索引类型，get_loc 有一些奇怪的属性，因此根据您的索引类型，您可能会返回一个索引，一个面具，或切片。这对我来说有点令人沮丧，因为我不想修改整个列只是为了提取一个变量的索引。更简单的是完全避免该功能：

list(df.columns).index('pear')

非常简单，而且可能相当快。

从python pandas中的列名获取列索引

8 个答案: