在 R 中,当您需要根据可以执行的列名称检索列索引时
idx <- which(names(my_data)==my_colum_name)
有没有办法对 pandas 数据帧做同样的事情?
最佳答案
当然,你可以使用 .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"])
),虽然我绝对可以看到你想要索引号的情况。
https://stackoverflow.com/questions/13021654/