Python - 在 df 中查找特定名称

在数据分析和数据处理中，我们经常需要从数据框（DataFrame）中查找特定的行或列。本文将介绍如何在Python中使用 Pandas 库中的数据框（DataFrame）中查找特定名称的行或列。

查找特定名称的行

我们可以使用 df.loc 方法查找 DataFrame 中特定名称的行。假设我们有一个名为 df 的 DataFrame，其中包含以下数据：

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({'Name': ['John', 'Mary', 'Tom'],
                   'Age': [25, 30, 35],
                   'City': ['New York', 'Los Angeles', 'Chicago']})

# 查看 DataFrame
print(df)

输出结果如下：

   Name  Age         City
0  John   25     New York
1  Mary   30  Los Angeles
2   Tom   35      Chicago

现在，我们想要查找 Name 为 Mary 的行，可以使用以下代码：

# 查找 Name 为 Mary 的行
row = df.loc[df['Name'] == 'Mary']
print(row)

输出结果如下：

   Name  Age         City
1  Mary   30  Los Angeles

查找特定名称的列

我们可以使用 DataFrame 的 filter 或 loc 方法来查找 DataFrame 中特定名称的列。假设我们仍然有一个名为 df 的 DataFrame，现在我们想要从 DataFrame 中选择名称为 Age 和 City 的列：

# 使用 filter 方法查找列
cols = df.filter(items=['Age', 'City'])
print(cols)

输出结果如下：

   Age         City
0   25     New York
1   30  Los Angeles
2   35      Chicago

或者，我们可以使用 loc 方法和布尔索引来选择列：

# 使用 loc 方法查找列
cols = df.loc[:, ['Age', 'City']]
print(cols)

输出结果与前一个示例相同：

   Age         City
0   25     New York
1   30  Los Angeles
2   35      Chicago

此外，我们也可以通过以下方式获取特定名称的列：

# 选择 Name 列
name = df['Name']
print(name)

输出结果如下：

0    John
1    Mary
2     Tom
Name: Name, dtype: object

以上是在 Python 中查找 DataFrame 中特定名称的行或列的方法。Pandas 库在数据处理和数据分析方面非常强大，值得我们深入学习。