Python编程：返回具有指定索引元素的行

在Python中，有时候我们需要获取某一个表格或文本文件中具有指定元素的行，完全遍历所有行很浪费时间，而且代码冗长。本文将介绍一种快速获取具有指定索引元素的行的方法。

示例数据

我们首先需要有一些数据用于演示。下面是一个文本文件test.txt的示例内容：

ID  Name    Gender  Age
1   Tom     Male    20
2   Mary    Female  22
3   John    Male    25
4   Lily    Female  21
5   Bob     Male    23

我们在代码中可以使用pandas库读取这个文本文件，并将其转换为数据框形式，示例代码如下：

import pandas as pd

df = pd.read_csv('test.txt', sep='\t')
print(df.head())

输出结果为：

   ID Name  Gender  Age
0   1  Tom    Male   20
1   2 Mary  Female   22
2   3 John    Male   25
3   4 Lily  Female   21
4   5  Bob    Male   23

返回具有指定索引元素的行

现在我们需要找到一个方法来返回具有指定索引元素的行，我们可以通过pandas库中的iloc方法实现。iloc方法接受两个参数，第一个参数表示返回具有指定索引元素的行，第二个参数表示需要返回的列数。

下面的示例代码演示了如何返回具有指定索引元素的行：

# 返回ID值为1的行的所有列的值
row1 = df.iloc[df.index[df['ID']==1].tolist()[0], :]
print(row1)

# 返回Gender值为Female的行的所有列的值
row2 = df.iloc[df.index[df['Gender']=='Female'].tolist(), :]
print(row2)

输出结果为：

ID            1
Name        Tom
Gender     Male
Age          20
Name: 0, dtype: object

   ID  Name  Gender  Age
1   2  Mary  Female   22
3   4  Lily  Female   21

上述示例代码中的第一段代码返回具有ID值为1的行的所有列的值，第二段代码返回具有Gender值为Female的行的所有列的值。