以下是一个示例代码,演示如何遍历数据框、收集数据并创建一个新的数据框:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['Tom', 'Nick', 'John'],
'Age': [25, 30, 35],
'City': ['New York', 'Paris', 'London']}
df = pd.DataFrame(data)
# 创建一个空的新数据框
new_df = pd.DataFrame(columns=['Name', 'City'])
# 遍历数据框
for index, row in df.iterrows():
if row['City'] == 'New York':
# 收集数据并添加到新数据框
new_df = new_df.append(row)
# 显示新数据框
print(new_df)
输出结果为:
Name City
0 Tom New York
在这个示例中,我们创建了一个名为df
的数据框,并遍历它的每一行。如果某一行的City
列的值为New York
,则将该行的数据添加到一个新的数据框new_df
中。最后,我们显示了新数据框的内容。