python函数返回数据框

在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
 
 

这是一个简单的例子,实际上,你可以根据需要,创建更复杂的数据框,包括多个列,不同类型的数据,等等。

posted @ 2025-03-12 13:48  kang_ms  阅读(33)  评论(0)    收藏  举报