如何使用 Dataframe 内容作为 function 输入？

Question

晚上的伙伴们，

我想知道是否有使用表格内容作为 function 的输入而不必获取每个单独的值。

就像我们为元组拆表一样。 有点函数（*（元组）），但我不想使用元组，而是想使用 dataframe 的单行，其中每一列将用作顺序输入。

我想做的是：

# Table that has a single row with name and age
table_content = table.loc[table['ID'] == id]

# I want to cast it like this
function_for_name_and_age(table_content)

# Without having to do
table_content_name = table_content['Name'].values(0)
table_content_age = table_content['Age'].values(0)

function_for_name_and_age(table_content_name, table_content_age )

'''

Answer 1

您可以将 df 行转换为元组。

import pandas as pd


table = pd.DataFrame([[1, "Tom", 36, 3], [2, "Bob", 30, 2]], columns=["ID", "Name", "Age", "Children"])
table_content = table.loc[table["ID"] == 1]

_id, name, age, children = table_content.to_records(index=False)[0]


def function_for_name_and_age(table_name: str, table_age: int) -> tuple:
    return table_name, table_age


print(function_for_name_and_age(name, age))