Python Pandas:如何检索多个值的行索引

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# 检索Age列中值为30和40的行索引
indexes = df[df['Age'].isin([30, 40])].index
print(indexes)

Int64Index([1, 3], dtype='int64')

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# 检索Age列中值为30和40的行索引
indexes = df.query('Age in [30, 40]').index
print(indexes)

Int64Index([1, 3], dtype='int64')

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# 检索Age列中值为30和40的行索引
indexes = df.loc[df['Age'].isin([30, 40])].index
print(indexes)

Int64Index([1, 3], dtype='int64')