在Pandas中循环数据帧并根据子串选择行

import pandas as pd

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

# 定义一个空列表，用于存储符合条件的行索引
selected_rows = []

# 循环遍历数据帧的每一行
for index, row in df.iterrows():
    # 判断某一列中是否包含指定的子串
    if 'ar' in row['Name']:
        # 将符合条件的行索引添加到列表中
        selected_rows.append(index)

# 根据选定的行索引，选择对应的行数据
selected_df = df.loc[selected_rows]

# 打印选择的行数据
print(selected_df)

      Name  Age    City
1      Bob   30   Paris
2  Charlie   35  London