您提供的代码有一些错误,我帮您做了一些修正:
import pandas as pd
def big_countries(world: pd.DataFrame) -> pd.DataFrame:
return world.loc[world['area'] >= 3000000, ['name', 'population', 'area']]
# 使用示例
df = pd.DataFrame({
'name': ['China', 'India', 'USA'],
'population': [1439323776, 1380004385, 331002651],
'area': [9640011, 3287590, 9629091]
})
result = big_countries(df)
print(result)
这个函数接受一个名为 world
的 Pandas DataFrame 对象作为参数,并返回一个包含大国名称、人口和面积的子集 DataFrame。如果国家的面积大于等于3000000,就将其包括在结果中。请注意,在代码中使用的是 loc 方法来选择符合条件的行。
使用示例展示了如何调用 big_countries
函数并传递一个测试数据集。结果将打印在控制台上。
希望对您有所帮助!如有任何疑问,请随时提出。
内容由零声教学AI助手提供,问题来源于学员提问