要按索引查询数据框,可以使用loc
方法。loc
方法接受一个索引列表作为参数,并返回对应索引的数据。
下面是一个示例代码:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['Tom', 'Nick', 'John', 'Alice'],
'Age': [26, 24, 28, 22],
'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data, index=['A', 'B', 'C', 'D'])
print(df)
# 按索引查询数据
index_list = ['A', 'C']
result = df.loc[index_list]
print(result)
输出:
Name Age City
A Tom 26 New York
B Nick 24 Paris
C John 28 London
D Alice 22 Tokyo
Name Age City
A Tom 26 New York
C John 28 London
在上面的示例中,我们首先创建了一个示例数据框df
,然后使用loc
方法按索引查询了索引为'A'和'C'的数据,并将结果赋值给result
变量。最后,我们打印了结果。注意,loc
方法返回的是一个新的数据框,包含指定索引的行数据。