在Python中,我们可以使用pandas库创建和返回数据框(DataFrame)。以下是一个函数示例,该函数返回一个简单的数据框:
import pandas as pd
def create_dataframe():
data = {'Name': ['Tom', 'Nick', 'John'], 'Age': [20, 21, 19]}
df = pd.DataFrame(data)
return df
在这个函数中,我们首先导入了pandas库,然后定义了一个名为 create_dataframe
的函数。在这个函数内部,我们定义了一个字典 data
,其中包含两个键:'Name'和'Age',每个键对应的值是一个列表。然后,我们使用 pd.DataFrame(data)
创建了一个数据框,并将其赋值给变量 df
。最后,我们返回了这个数据框。
你可以通过以下方式调用这个函数并打印返回的数据框:
df = create_dataframe()
print(df)
运行上述代码,你将看到如下输出:
Name Age
0 Tom 20
1 Nick 21
2 John 19
这是一个简单的例子,实际上,你可以根据需要,创建更复杂的数据框,包括多个列,不同类型的数据,等等。