将基于标签的索引(loc)转换为基于位置的索引(iloc)

2024-04-26 00:07:12 发布

您现在位置:Python中文网/ 问答频道 /正文

根据我的数据:

import numpy as np
import pandas as pd

data = pd.DataFrame(
    {
        'x': [11, 12, 13],
        'y': [21, 22, 23]
    },
    index=['a', 'b', 'c']
)

我想找到与标签['a', 'c']对应的位置索引。通过目视检查,我知道它们是[0, 2],但这仅在玩具示例中可能

“暴力”方法是

my_lab = ['a', 'c']
my_pos = (data.index == my_idx).to_numpy().nonzero()[0]

有没有更好/更快/更优雅的方法?还是暴力手段是唯一的手段


Tags: 数据方法importnumpydataframepandasdataindex

热门问题